vampire.analysis.cluster_contours

vampire.analysis.cluster_contours(pc, contours, num_clusters=5, num_pc=20, random_state=None)

K-means clustering of contour principal components.

Parameters

pcndarray

Principal components of contours.

contoursndarray

Object contours, with shape (num_contour, 2*num_points).

num_clustersint, optional

Number of clusters.

num_pcint, optional

Number of principal components used for approximation.

random_stateNone or int, optional

Random state for K-means clustering.

Returns

contours_dfDataFrameDataFrame of objects’ contour coordinates with cluster id.

centroidsndarray

Coordinates of cluster centers of K-means clusters.

See also

sklearn.cluster.KMeans

Implementation of K-means clustering.

vampire.analysis.pca_transform_contours vampire.analysis.assign_clusters_id