tf2 history keys

dmbee · Nov 8, 2019 · 8d2d067 · 8d2d067
1 parent e72ad1a
commit 8d2d067
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
 
 ##############################################