pino-source-code / scripts /find_split_sweep.py
mattbitzesty's picture
feat(scripts): add dataset split validation utilities
d03a111 unverified
Raw
History Blame Contribute Delete
297 Bytes
from pino.upload_data import create_molecule_disjoint_split
for ratio in [0.4, 0.45, 0.5, 0.55, 0.6]:
dd = create_molecule_disjoint_split('data/synthetic_dataset_v2.jsonl', train_ratio=ratio, seed=42)
print("ratio=%.2f train=%d val=%d" % (ratio, len(dd['train']), len(dd['validation'])))