Skip to content

Commit bff97ad

Browse files
committed
Adapted ut1 to FFNN changes
1 parent 97704f4 commit bff97ad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/ut1/main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,8 +154,8 @@ int main(){
154154
FeedForwardNeuralNetwork * ffnn = new FeedForwardNeuralNetwork(2, 2, 2);
155155
GaussianActivationFunction * gss_actf = new GaussianActivationFunction();
156156
IdentityActivationFunction * id_actf = new IdentityActivationFunction();
157-
ffnn->setNNLayerActivationFunction(0, gss_actf);
158-
ffnn->setNNLayerActivationFunction(1, id_actf);
157+
ffnn->getNNLayer(0)->setActivationFunction(gss_actf);
158+
ffnn->getNNLayer(1)->setActivationFunction(id_actf);
159159
ffnn->connectFFNN();
160160
ffnn->assignVariationalParameters(); // make all betas variational
161161

0 commit comments

Comments
 (0)