Using right model variables for prediction in the multiclass classification examples (sec 3.1)?

[robjm16]

Great tutorial and notebooks! Thanks.

I think the model variables in the multiclass classification visualization section need to be updated from just "model" to "sr_model" (first snippet) and "deep_model" (second snippet) below:

y_pred_class = model.predict_classes(X_test, verbose=0)
y_test_class = np.argmax(y_test, axis=1)
print(classification_report(y_test_class, y_pred_class))
plot_confusion_matrix(model, X_test, y_test_class)

y_pred_class = model.predict_classes(X_test, verbose=0)
y_test_class = np.argmax(y_test, axis=1)
print(classification_report(y_test_class, y_pred_class))
plot_confusion_matrix(model, X_test, y_test_class)