Hi Guys,
I am new to PyTorch and I need help. I was trying to use the sample code here: https://github.com/spdin/time-series-prediction-lstm-pytorch/blob/master/Time_Series_Prediction_with_LSTM_Using_PyTorch.ipynb?fbclid=IwAR0pzLvhdhz0kl2_tvJEGAybzRoXuIFlzUOpCxSHUzP-slWh6RIgHhbBUV0 6
it works well, but when I tried to add more input features I got Error in the testing section for the following lines:
data_predict = sc.inverse_transform(data_predict)
dataY_plot = sc.inverse_transform(dataY_plot)

The error says “non-broadcastable output operand with shape (110,1) doesn’t match the broadcast shape”
so the model works well, but I can’t retrieve the original value of Y. So what else Should I modify in the code? I modified  “input_size” in training section and added so following line in Data loading section:
y = (y[:,0]).reshape(y[:,0].shape[0],1) 

and finally
training_set = df.iloc[:,1:3].values

Thanks in advance