search

a-r-j / ProteinWorkshop

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)
https://proteins.sh/
MIT License
194 stars 16 forks source link

changed time_cutoff option #89

Closed kierandidi closed 6 months ago

kierandidi commented 6 months ago

@a-r-j

Added option for the PDBDataModule to create splits based on time interval by exposing PDBManager functionality.

Tested both this functionality as well as random and sequence_similarity to ensure no regression is happening

amorehead commented 6 months ago

LGTM

a-r-j commented 6 months ago

Ah, some tests are failing @kierandidi

kierandidi commented 6 months ago

should be fixed now @a-r-j