Skip to content

Commit 61235de

Browse files
Changes for latest api
1 parent acee25a commit 61235de

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/main/java/org/ml4j/nn/models/inceptionv4/impl/DefaultInceptionV4Factory.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ public SupervisedFeedForwardNeuralNetwork createInceptionV4(FeedForwardNeuralNet
9292

9393
return sessionFactory
9494
.createSession(trainingContext.getDirectedComponentsContext())
95-
.buildNeuralNetwork("inceptionV4", inceptionV4Definition.getInputNeurons())
95+
.buildSupervised3DNeuralNetwork("inceptionV4", inceptionV4Definition.getInputNeurons())
9696
.withComponentGraphDefinition(inceptionV4Definition)
9797
.build();
9898
}
@@ -112,7 +112,7 @@ public SupervisedFeedForwardNeuralNetwork createInceptionV4(FeedForwardNeuralNet
112112

113113
return sessionFactory
114114
.createSession(trainingContext.getDirectedComponentsContext())
115-
.buildNeuralNetwork("inceptionV4WithRegularisation", inceptionV4Definition.getInputNeurons())
115+
.buildSupervised3DNeuralNetwork("inceptionV4WithRegularisation", inceptionV4Definition.getInputNeurons())
116116
.withComponentGraphDefinition(inceptionV4Definition)
117117
.build();
118118

@@ -132,7 +132,7 @@ public SupervisedFeedForwardNeuralNetwork createInceptionV4WithCustomTail(FeedFo
132132

133133
return sessionFactory
134134
.createSession(trainingContext.getDirectedComponentsContext())
135-
.buildNeuralNetwork("inceptionV4WithCustomTail", inceptionV4Definition.getInputNeurons())
135+
.buildSupervised3DNeuralNetwork("inceptionV4WithCustomTail", inceptionV4Definition.getInputNeurons())
136136
.withComponentGraphDefinition(inceptionV4Definition)
137137
.build();
138138
}
@@ -149,7 +149,7 @@ public SupervisedFeedForwardNeuralNetwork createInceptionV4Tail(FeedForwardNeura
149149

150150
return sessionFactory
151151
.createSession(trainingContext.getDirectedComponentsContext())
152-
.buildNeuralNetwork("inceptionV4CustomTail", inceptionV4Definition.getInputNeurons())
152+
.buildSupervised3DNeuralNetwork("inceptionV4CustomTail", inceptionV4Definition.getInputNeurons())
153153
.withComponentGraphDefinition(inceptionV4Definition)
154154
.build();
155155
}
@@ -165,7 +165,7 @@ public SupervisedFeedForwardNeuralNetwork createInceptionV4WithoutTail(FeedForwa
165165

166166
return sessionFactory
167167
.createSession(trainingContext.getDirectedComponentsContext())
168-
.buildNeuralNetwork("inceptionV4CustomTail", inceptionV4Definition.getInputNeurons())
168+
.buildSupervised3DNeuralNetwork("inceptionV4CustomTail", inceptionV4Definition.getInputNeurons())
169169
.withComponentGraphDefinition(inceptionV4Definition)
170170
.build();
171171
}

0 commit comments

Comments
 (0)