plot_decision_boundary for model 7 reads "doing multiclass classification"

[pekasus]

In trying to visualize the metrics for model_7, I ran plot_boundary_decision(model_7, X, y) and it prints doing multiclass classification. I've tried adding the input_shape parameter to the first layer, as well as expand_dims in the model_7.fit function call, and these result in errors that logits and labels have to be the same dimension.

The plot drawn is all orange and doesn't look like the video, but the data in the model is the same as the video.

[pekasus]

I figured out what was wrong. In the function, i was looking at the len(y_pred) instead of len(y_pred[0]).

The graph is working now.

[mrdbourke]

Glad to hear you got it working @pekasus !

Thank you for updating your issue.