diff --git a/tests/test-callbacks.cpp b/tests/test-callbacks.cpp index ed4c9b4..b17e854 100644 --- a/tests/test-callbacks.cpp +++ b/tests/test-callbacks.cpp @@ -2,6 +2,7 @@ #include #include #include +#include #include #include @@ -11,7 +12,7 @@ TEST_CASE( "EarlyStopping callback throws exception when the metric is not found", "[callback]") { std::shared_ptr earlyStopping = - std::make_shared("LOSS", 0.1); + std::make_shared("NOT_A_METRIC", 0.1); Network network; diff --git a/tests/test-network.cpp b/tests/test-network.cpp index e6c6ac7..a6ef157 100644 --- a/tests/test-network.cpp +++ b/tests/test-network.cpp @@ -214,7 +214,7 @@ SCENARIO("The network updates the weights and biases as pre-calculated") { Network checkpoint; - Model::load_from_file("checkpoint-0.bin", checkpoint); + Model::load_from_file("N9NeuralNet7NetworkE-checkpoint-0.bin", checkpoint); REQUIRE(checkpoint.getNumLayers() == network.getNumLayers());