update head

mlpack · Apr 3, 2018 · 00e2c9f · 00e2c9f
1 parent b3451ab
commit 00e2c9f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp
@@ -647,8 +647,9 @@ BOOST_AUTO_TEST_CASE(GradientFlexibleReLULayerTest)
       input = arma::randn(10, 1);
       target = arma::mat("1");
 
-      model = new FFN<NegativeLogLikelihood<>, NguyenWidrowInitialization>(
-        input, target);
+      model = new FFN<NegativeLogLikelihood<>, NguyenWidrowInitialization>();
+      model->Predictors() = input;
+      model->Responses() = target;
       model->Add<Linear<> >(10, 2);
       model->Add<FlexibleReLU<> >(0.05);
       model->Add<LogSoftMax<> >();