Custom dataset. Data preprocessing

[Chicky-Picky]

Thank you so much for your outstanding work!

I'm interested in your models and would like to run them on some custom datasets. Unfortunately, I haven't found any instructions on how to preprocess the raw data. Could you please tell me whether it is possible to run your models on custom datasets? And if so, where can I find your preprocessing script?

Thank you!

[Oxer11]

Hi!

To define your custom datasets, you can refer to this page. Basicly, you need to define how to load your dataset with the data.ProteinDataset. load_pdbs function and save it as a pickle file. Also, you need to define how to load the labels and get the items.