tensor creation KeyError: 5

[zaineb]

For lesson 05-linear-model-and-neural-net-from-scratch.ipynb the above command gives an exception (KeyError: 5) on kaggle. Need to use .values as df.Survived.values for this to work correctly. Kindly update.

Thanks