Showing
1 changed file
with
4 additions
and
0 deletions
... | @@ -195,6 +195,10 @@ if __name__ == "__main__": | ... | @@ -195,6 +195,10 @@ if __name__ == "__main__": |
195 | history = model.fit(train_features, train_labels, | 195 | history = model.fit(train_features, train_labels, |
196 | epochs=50, verbose=0, validation_split=0.25) | 196 | epochs=50, verbose=0, validation_split=0.25) |
197 | 197 | ||
198 | + # Print model metrics | ||
199 | + print("Model metics:") | ||
200 | + print("\n\t%s:" % (model.metrics_names[1])) | ||
201 | + | ||
198 | # Plot training-validation loss | 202 | # Plot training-validation loss |
199 | plt.figure() | 203 | plt.figure() |
200 | plt.plot(history.history['loss']) | 204 | plt.plot(history.history['loss']) | ... | ... |
-
Please register or login to post a comment