Issues in preprocess.py

[taein98]

Thanks for your nice work.

I think I found a bug in preprocessing code for the reaction class unknown set. (e.g. --use_class option preprocess.py)
When I use the --use_class option in preprocess.py c for reaction class unknown training(as its action is stoe_false), it saves the dataset like below.

And when I load the dataset.pkl file in train_center.py
It always load the batch_data with the below form.

It produces the error in reaction class unknown training.

I believe that the preprocess.py code change like

Wait for your answer. Thank you

[ziqi92]

Sorry for the late reply!

Thank you for your suggestion! I will check the error and get back to you in these days!

Best,
Ziqi

[ziqi92]

Thank you again for pointing out the issue in our preprocess file! I really appreciate it!

I have changed the code as you suggested.

Best,
Ziqi