mattertune.data.atoms_list

Classes

AtomsListDataset(config)

AtomsListDatasetConfig(*[, type])

class mattertune.data.atoms_list.AtomsListDatasetConfig(*, type='atoms_list', atoms_list)[source]
Parameters:
  • type (Literal['atoms_list'])

  • atoms_list (list[Atoms])

type: Literal['atoms_list']

Discriminator for the atoms_list dataset.

atoms_list: list[ase.Atoms]

The list of Atoms objects.

create_dataset()[source]
class mattertune.data.atoms_list.AtomsListDataset(config)[source]
Parameters:

config (AtomsListDatasetConfig)

__init__(config)[source]
Parameters:

config (AtomsListDatasetConfig)