espaloma.data.utils

Functions

batch(ds, batch_size[, seed])

Batch graphs and values after shuffling.

collate_fn(graphs)

from_csv(path[, toolkit, smiles_col, ...])

Read csv from file.

infer_mol_from_coordinates(coordinates, species)

make_temp_directory()

normalize(ds)

Get mean and std.

split(ds, partition)

Split the dataset according to some partition.

sum_offsets(elements)