Carlos-Francisco Méndez-Cruz / deep-learning-workshop

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
959959c2 1 parent 5ca9b829

Deep Learning Workshop

Inline Side-by-side
Showing 1 changed file with 4 additions and 0 deletions
......@@ -195,6 +195,10 @@ if __name__ == "__main__":
history = model.fit(train_features, train_labels,
epochs=50, verbose=0, validation_split=0.25)
# Print model metrics
print("Model metics:")
print("\n\t%s:" % (model.metrics_names[1]))
# Plot training-validation loss
plt.figure()
plt.plot(history.history['loss'])
......