Group Detection¶

class symdet.symmetry_group_extraction.group_detection.GroupDetection(model: DenseModel, data_clusters: dict, representation_set: str = 'train')[source]¶

A class to cluster raw data into symmetry groups.

model¶

Model to use in the group detection.

Type: DenseModel

data_clusters¶

Data cluster class used for the partitioning of the data.

Type: dict

representation_set¶

Which set to use in the representation, train, validation, or test.

Type: str

Methods

run_symmetry_detection([plot, save])

Run the symmetry detection routine.

run_symmetry_detection(plot: bool = True, save: bool = False)[source]¶

Run the symmetry detection routine.

Parameters

plot (bool) – Plot the TSNE visualization.
save (bool) – Save the image plotted.

Return type

None