stratified_dataset_split

autoplex.data.common.utils.stratified_dataset_split(atoms, split_ratio)

Split the dataset.

Parameters:

• atoms (Atoms) – ase Atoms object

• split_ratio (float) – Parameter to divide the training set and the test set.

Returns:

split-up datasets of train structures and test structures.

Return type:

train_structures, test_structures