nfflr.data.deepmd_hea_dataset

nfflr.data.deepmd_hea_dataset#

nfflr.data.deepmd_hea_dataset(datadir: Path | str, n_train: float | int = 0.9, n_val: float | int = 0.1, transform: Module | None = None, diskcache: bool = False)[source]#

Load deepmd numpy format dataset.