diff --git a/src/examples/net_test_1.cpp b/src/examples/net_test_1.cpp index 33451e6c3c02b1feeacfe9390699a8ac0935f21d..372fa3dcffb42800e10fed0eeb63f6e3be64e8b4 100644 --- a/src/examples/net_test_1.cpp +++ b/src/examples/net_test_1.cpp @@ -141,5 +141,10 @@ int main() { net.randomize_parameters(); optimize_via_gradient_descent( net, mse ); + /* Normalize data to prevent 'nan' results */ + ds.normalize(); + net.randomize_parameters(); + optimize_via_gradient_descent(net, mse); + return 0; }