How can i add a new dataset?

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.

MIT License

2.08k stars 334 forks source link

How can i add a new dataset? #26

Closed AmeenAli closed 2 years ago

AmeenAli commented 2 years ago

how can i add a new dataset for classification with classes > 2 ?

zhengsx commented 2 years ago

You may add your own dataset from here, and set num_class = <classes>. For classification problem with multiple classes, cross entropy loss is recommended.

zhengsx commented 2 years ago

Close this issue due to inactive. Feel free to raise a new one or reopen this one for any further question.