data.NumberOfAtomsCriterion

class data.NumberOfAtomsCriterion[source]

A callable class that returns the number of atoms for each sample in the dataset.