diff --git a/examples/03_linear_regression_sol.py b/examples/03_linear_regression_sol.py index 6fde7360..3705ef98 100644 --- a/examples/03_linear_regression_sol.py +++ b/examples/03_linear_regression_sol.py @@ -48,7 +48,7 @@ writer = tf.summary.FileWriter('./graphs/linear_reg', sess.graph) # Step 8: train the model - for i in range(50): # train the model 100 epochs + for i in range(100): # train the model 100 epochs total_loss = 0 for x, y in data: # Session runs train_op and fetch values of loss @@ -67,4 +67,4 @@ plt.plot(X, Y, 'bo', label='Real data') plt.plot(X, X * w + b, 'r', label='Predicted data') plt.legend() -plt.show() \ No newline at end of file +plt.show()