core.scripts.create_finetune_dataset#
Attributes#
Functions#
|
Given a path to an ASE database file, compute the normalizer value and linear |
Extract energy and forces from an ASE atoms object at a given index in the dataset. |
|
|
Compute linear reference coefficients given atomic numbers and energies. |
|
Write ASE atoms objects to an ASE database file. This function is designed to be |
|
Driver script to launch processing of ASE atoms files into an ASE database. |
Module Contents#
- core.scripts.create_finetune_dataset.compute_normalizer_and_linear_reference(train_path, num_workers)#
Given a path to an ASE database file, compute the normalizer value and linear reference coefficients. These are used to normalize energies and forces during training. For large datasets, compute this for only a subset of the data.
- core.scripts.create_finetune_dataset.extract_energy_and_forces(idx)#
Extract energy and forces from an ASE atoms object at a given index in the dataset.
- core.scripts.create_finetune_dataset.compute_lin_ref(atomic_numbers, energies)#
Compute linear reference coefficients given atomic numbers and energies.
- core.scripts.create_finetune_dataset.write_ase_db(mp_arg)#
Write ASE atoms objects to an ASE database file. This function is designed to be run in parallel using multiprocessing.
- core.scripts.create_finetune_dataset.launch_processing(data_dir, output_dir, num_workers)#
Driver script to launch processing of ASE atoms files into an ASE database.
- core.scripts.create_finetune_dataset.parser#