diff --git a/mlp.py b/mlp.py index cf892959dc561a2ebafe94ea42aa44754c2c51dc..ac45a1fe9735eefd1357c2d3e46fc527f6389829 100644 --- a/mlp.py +++ b/mlp.py @@ -161,7 +161,7 @@ if __name__== '__main__': train_accuracies, test_accuracy=run_mlp_training(data_train, labels_train, data_test, labels_test,d_h,learning_rate,num_epoch) train_accuracies.append(test_accuracy) - print(train_accuracies) + K=list(range(num_epoch+1)) plt.plot(K,train_accuracies) plt.title("Accuracy=f(epoch)")