datasets.ISO17
- class datasets.ISO17(*args: Any, **kwargs: Any)[source]
ISO17 benchmark data set for molecular dynamics of C7O2H10 isomers containing molecular forces.
References:
- Parameters:
datapath – path to dataset
fold – select a specific dataset of iso17
batch_size – (train) batch size
num_train – number of training examples
num_val – number of validation examples
num_test – number of test examples
split_file – path to npz file with data partitions
format – dataset format
load_properties – subset of properties to load
val_batch_size – validation batch size. If None, use test_batch_size, then batch_size.
test_batch_size – test batch size. If None, use val_batch_size, then batch_size.
transforms – Transform applied to each system separately before batching.
train_transforms – Overrides transform_fn for training.
val_transforms – Overrides transform_fn for validation.
test_transforms – Overrides transform_fn for testing.
num_workers – Number of data loader workers.
num_val_workers – Number of validation data loader workers (overrides num_workers).
num_test_workers – Number of test data loader workers (overrides num_workers).
distance_unit – Unit of the atom positions and cell as a string (Ang, Bohr, …).