Cluster expansions¶
- class icet.ClusterExpansion(cluster_space, parameters, metadata=None)[source]
Cluster expansions are obtained by combining a cluster space with a set of parameters, where the latter is commonly obtained by optimization. Instances of this class allow one to predict the property of interest for a given structure.
Note: Each element of the parameter vector corresponds to an effective cluster interaction (ECI) multiplied by the multiplicity of the underlying orbit.
- cluster_space
cluster space that was used for constructing the cluster expansion
- Type
- parameters
parameter vector
- Type
np.ndarray
Example
The following snippet illustrates the initialization and usage of a ClusterExpansion object. Here, the parameters are taken to be a list of ones. Usually, they would be obtained by training with respect to a set of reference data:
>>> from ase.build import bulk >>> from icet import ClusterSpace, ClusterExpansion >>> # create cluster expansion with fake parameters >>> prim = bulk('Au') >>> cs = ClusterSpace(prim, cutoffs=[7.0, 5.0], ... chemical_symbols=[['Au', 'Pd']]) >>> parameters = len(cs) * [1.0] >>> ce = ClusterExpansion(cs, parameters) >>> # make prediction for supercell >>> sc = prim.repeat(3) >>> for k in [1, 4, 7]: >>> sc[k].symbol = 'Pd' >>> print(ce.predict(sc))
- property fractional_position_tolerance: float
tolerance applied when comparing positions in fractional coordinates (inherited from the underlying cluster space)
- Return type
float
- get_cluster_space_copy()[source]
Gets copy of cluster space on which cluster expansion is based
- Return type
- property metadata: dict
metadata associated with cluster expansion
- Return type
dict
- property orders: List[int]
orders included in cluster expansion
- Return type
List
[int
]
- property parameters: List[float]
parameter vector; each element of the parameter vector corresponds to an effective cluster interaction (ECI) multiplied by the multiplicity of the respective orbit
- Return type
List
[float
]
- property position_tolerance: float
tolerance applied when comparing positions in Cartesian coordinates (inherited from the underlying cluster space)
- Return type
float
- predict(structure)[source]
Predicts the property of interest (e.g., the energy) for the input structure using the cluster expansion.
- Parameters
structure (
Union
[Atoms
,Structure
]) – atomic configuration- Returns
property value of predicted by the cluster expansion
- Return type
float
- property primitive_structure: Atoms
primitive structure on which cluster expansion is based
- Return type
- print_overview(print_threshold=None, print_minimum=10)[source]
Print an overview of the cluster expansion in terms of the orbits (order, radius, multiplicity, corresponding ECI etc).
- Parameters
print_threshold (
Optional
[int
]) – if the number of orbits exceeds this number print dotsprint_minimum (
int
) – number of lines printed from the top and the bottom of the orbit list if print_threshold is exceeded
- Return type
None
- prune(indices=None, tol=0)[source]
Removes orbits from the cluster expansion (CE), for which the absolute values of the corresponding parameters are zero or close to zero. This commonly reduces the computational cost for evaluating the CE and is therefore recommended prior to using it in production. If the method is called without arguments orbits will be pruned, for which the ECIs are strictly zero. Less restrictive pruning can be achieved by setting the tol keyword.
- Parameters
indices (
Optional
[List
[int
]]) – indices of parameters to remove from the cluster expansion.tol (
float
) – all orbits will be pruned for which the absolute parameter value(s) is/are within this tolerance
- Return type
None
- static read(filename)[source]
Reads ClusterExpansion object from file.
- Parameters
filename (
str
) – file from which to read
- property symprec: float
tolerance imposed when analyzing the symmetry using spglib (inherited from the underlying cluster space)
- Return type
float
- to_dataframe()[source]
Returns representation of the cluster expansion in the form of a DataFrame containing orbit information and effective cluster interactions (ECIs).
- Return type
- write(filename)[source]
Writes ClusterExpansion object to file.
- Parameters
filename (
str
) – name of file to which to write- Return type
None