cur_select

Contents

cur_select#

autoplex.data.common.utils.cur_select(atoms, selected_descriptor, kernel_exp, select_nums, stochastic=True, random_seed=None)[source]#

Perform CUR selection on a set of atoms to get representative SOAP descriptors.

Parameters:
  • atoms (list of ase.Atoms) – The atoms for which to perform CUR selection.

  • selected_descriptor (str) – The quip descriptor string to use for the calculation.

  • kernel_exp (float) – The kernel exponent to use in the calculation.

  • select_nums (int) – The number of atoms to select.

  • stochastic (bool) – Whether to perform stochastic CUR selection.

  • random_seed (int) – The seed for the random number generator.

Returns:

The selected atoms.

Return type:

list of ase.Atoms

Notes

This function calculates the descriptor vector for each atom, then performs CUR selection on the resulting vectors.

References