Skip to content

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

Description

@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).

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions