01_PyTorch Workflow - 42. timestamp 5:35:00 - Problem with plot_prediction

[VELTAX]

Hi, i need some help, here my problem:
when trying to create the prediction graphic, i run into a error:
.

I understand that the problem is that my x and Y are not the same, so i tried to search why, and couldnt find, i have all the same value as in the video, but get more prediction from model_0(X_test)
.

[sudokit]

Can you send the model?

[VELTAX]

Here the model:

[sudokit]

Hmm thought u would have a wrong output shape... Weird.

[TiagoNFC]

The problem seems to be in :

def forward(self, x: X_test) -> torch.tensor

it should read:

def forward(self, X: torch.tensor) -> torch.tensor

[VELTAX]

i've found the problem ome timz ago and forgot to close this discussion, the problem was in the model at:
return self.weights * X + self.bias the X was in upper-cas should have been in lower case : return self.weights * x + self.bias