Try to run it on google colab/ jupyter notebook, report error in "final_input = torch.cat((term_NN_out_map[self.root], drug_out), 1)" (line 177 of drugcell_NN.py))

[mqzhanglab]

For GPU based python 3.6.9. google colab

It will report error in last line(train_model) train_drugcell.py for the line 177 of drugcell_NN.py
Something about the root:

final_input = torch.cat((term_NN_out_map[self.root], drug_out), 1)

[jisoopark]

Hi,

Can you try to run the script on the command line and let me whether it still fails? Did you run all the functions in the "main" function (not specifically defined but placed after the train_drugcell function)?. I am little puzzled about the error because your ontology seems loaded correctly - there might be a problem in the initialization of term_NN_out_map.

Thanks,
Jisoo