diff --git a/examples/plot_nn_training_curves.py b/examples/plot_nn_training_curves.py index 29230d2..5903de8 100644 --- a/examples/plot_nn_training_curves.py +++ b/examples/plot_nn_training_curves.py @@ -70,8 +70,8 @@ def crnn_model(width=100, n_vars=6, n_classes=7, conv_kernel_size=5, pipe.fit(X_train, y_train) print(DataFrame(pipe.history.history)) -ac_train = pipe.history.history['acc'] -ac_val = pipe.history.history['val_acc'] +ac_train = pipe.history.history['accuracy'] +ac_val = pipe.history.history['val_accuracy'] epoch = np.arange(len(ac_train)) + 1 ##############################################