diff --git a/src/examples/main.cpp b/src/examples/main.cpp index 1bccc6548512d66606a13d7a17ca8ee416006d17..27b3ac53b1791ae825b805feb122f7b90a2257a6 100644 --- a/src/examples/main.cpp +++ b/src/examples/main.cpp @@ -53,13 +53,13 @@ int main(int argc, char** argv){ // 0.7, // 600, // 1000); - l4n::GradientDescent gs(10, 100); + l4n::GradientDescent gs(1e-3, 100); nn.randomize_weights(); /* Cross - validation */ l4n::CrossValidator cv(&gs, &mse); - cv.run_k_fold_test(1e-3, 100); + cv.run_k_fold_test(3, 10); /* Save network to the file */ nn.save_text("test_net.4n");