Error in the predict() function from the Report jupyter notebook..

[krishnanpc]

Hello,

Thanks for the great project. It is quite interesting. As a learning exercise, I was trying to reproduce your results by rerunning your project. (New to Pytorch)

I am trying to run your

Report_Thomas_MacDougall.ipynb

in colab.

During the predict phase in the Edge Memory NN,
I get a ValueError: Wrong number of items passed 2, placement implies 1 while trying to insert the output into the dataframe.

df.insert(1, 'pred_log_std_scaled', output, True).

When I look for the shape of the output variable in the predict function, it is of 46 x 2 and the test dataset is of 819 x 1. This size match might be causing the error. Just wondering what output means there? Should it also be of 819 x 1. ( ie; one value of each of the input entry).