Shape of test sample is different in YT video and notebbok on https://www.learnpytorch.io/02_pytorch_classification/

[Durga-33]

Video: 02. PyTorch Neural Network Classification
Timestamp: 9:54:07

Output in YT video
Length of test samples: 200 and Shape: torch.size ([200,2])

output in
https://www.learnpytorch.io/02_pytorch_classification/
shows

Length of predictions: 200, Shape: torch.Size([200, 1])
Length of test samples: 200, Shape: torch.Size([200])

Also why there is so much difference in predictions values given in YT and given in notebook

Please clarify

[mrdbourke]

Hi @Durga-33,

Good question(s)!

You're right about the shapes being different in the videos/notebooks.

I'd pay more attention to the notebooks, the videos have some mistakes in them but the notebooks are checked regularly to make sure they work.

I've just rerun the notebook you're talking about and it works for me.

If the random numbers are very different, make sure the seeds you're setting are the same as in the videos/notebooks.

Otherwise, small differences are to be expected depending on the computing platform you're using.

[Durga-33]

Okay Sir
Thank you