mattertune.data.xyz

Classes

XYZDataset(config)
XYZDatasetConfig(*[, type, down_sample, ...])

class mattertune.data.xyz.XYZDatasetConfig(*, type='xyz', src, down_sample=None, down_sample_refill=False)

Parameters:

• type (Literal['xyz'])

• src (str | Path)

• down_sample (int | None)

• down_sample_refill (bool)

type: Literal['xyz']

Discriminator for the XYZ dataset.

src: str | Path

The path to the XYZ dataset.

down_sample: int | None

Down sample the dataset

down_sample_refill: bool

Refill the dataset after down sampling to achieve the same length as the original dataset

create_dataset()

class mattertune.data.xyz.XYZDataset(config)

Parameters:

config (XYZDatasetConfig)

__init__(config)

Parameters:

config (XYZDatasetConfig)