01_pytorch_workflow_video.ipynb : generating plot predictions

[rnainggo]

Hello I just started to learn PyTorch and I tried to run the codes below as explained in the video :

def plot_preidctions(train_data = X_train,
                      train_labels = y_train,
                      test_data = X_test,
                      test_labels = y_test,
                      predictions = None) :
  """
  Plots training data, test data and compares predictions.
  """
  plt.figure(figsize=(10,7))
  plt.scatter(train_data, train_labels, c = "b", s = 4, label = "Training data")
  

  plt.scatter(test_data, test_labels, c = "g", s = 4, label = "Testing data")


  if predictions is not None:

    plt.scatter(test_data, predictions, c="r", s = 4, label = "Predictions")


  plt.legend(prop = {"size":14});

plot_predictions();

However after running plot_predictions();, it returns the following error

Please kindly advice what caused this and how I can resolve this error

[mrdbourke]

Hey @rnainggo,

What have you tried so far?

The error is saying that the function plot_predictions() is not available, have you tried to define it?

From notebook 01: https://github.com/mrdbourke/pytorch-deep-learning/blob/main/01_pytorch_workflow.ipynb

def plot_predictions(train_data=X_train, 
                     train_labels=y_train, 
                     test_data=X_test, 
                     test_labels=y_test, 
                     predictions=None):
  """
  Plots training data, test data and compares predictions.
  """
  plt.figure(figsize=(10, 7))

  # Plot training data in blue
  plt.scatter(train_data, train_labels, c="b", s=4, label="Training data")
  
  # Plot test data in green
  plt.scatter(test_data, test_labels, c="g", s=4, label="Testing data")

  if predictions is not None:
    # Plot the predictions in red (predictions were made on the test data)
    plt.scatter(test_data, predictions, c="r", s=4, label="Predictions")

  # Show the legend
  plt.legend(prop={"size": 14});

[ozzyoz14]

Spelling
def plot_preidctions
plot_predictions

[mrdbourke]

Ahh! That happens to me all the time! Haha

Especially in Google Colab