Skip to content

Add ZINC dataset, add limit to HF iterable datasets - #54

Merged
karinazad merged 4 commits into
mainfrom
zinc-dataset
Mar 26, 2025
Merged

Add ZINC dataset, add limit to HF iterable datasets#54
karinazad merged 4 commits into
mainfrom
zinc-dataset

Conversation

@karinazad

Copy link
Copy Markdown
Collaborator

No description provided.

@cgrambow cgrambow left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🚀

name="ZINC",
dataset_class=ZINCIterableDataset,
modality=Modality.SMILES,
supported_splits={Split.TRAIN, Split.TEST},

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

do we need train, val, test for supported splits for everything?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I guess it depends on whether we want to test on these datasets or on completely new data sources?

Comment thread src/lobster/data/_ume_datamodule.py Outdated
Comment thread src/lobster/datasets/_zinc_dataset.py
@karinazad
karinazad merged commit d43709c into main Mar 26, 2025
@karinazad
karinazad deleted the zinc-dataset branch March 26, 2025 00:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants