datamol-io / splito

Machine Learning dataset splitting for life sciences.
https://splito-docs.datamol.io/
Apache License 2.0
20 stars 2 forks source link

Added a Scikit-learn like `train_test_split` method #12

Closed cwognum closed 3 months ago

cwognum commented 3 months ago

Changelogs


Checklist:


You can now do:

from splito import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, smiles=smiles, method="scaffold")