My bad. The system is not allowing me to reply with a share line cuz they think it is promotional. Can you comment line 10 and 11 and change line 13 and 14 to:

x_train = torch.rand([5198,22])#d[‘mushrooms’][0][0:5198]
y_train = torch.rand([5198,1])#d[‘mushrooms’][1][0:5198].view(5198,-1)