snap-stanford / ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning
https://ogb.stanford.edu
MIT License
1.89k stars 397 forks source link

Add padding_index to AtomEncoder and BondEncoder #236

Closed HannesStark closed 3 years ago

HannesStark commented 3 years ago

Add padding_index to AtomEncoder and BondEncoder in mol_encoder.py to support, e.g., virtual edges with zero-padding as features

HannesStark commented 3 years ago

Pull request has problems that I need to fix first.