Helper Function not behaving as expected

[ravixteja]

Hey, in the 4th module - PyTorch Neural Networks CLassification I'm facing a problem with the helper function used to plot the decision boundary (Video 80).

So I downloaded the helper function and made it a module to be used anywhere in the directory and used it to plot. But There's no decision boundary plotted. Is this because of my models predictions ? Or is the function call wrong ? Please help

[NabJa]

A few questions:

1. Have you trained your model_0 before plotting? For info about how to train check out here.
2. Can you share the implementation of the model? The exact architecture.

If you haven't trained the model before plotting, this might be the issue.