pupil.models package

pupil.models.clustering module

class pupil.models.FaissKMeansClustering(n_clusters: int, n_init: int = 10, max_iter: int = 100)

Bases: object

distance_to_cluster_centers(X: NDArray2D) → Tuple[NDArray2D, NDArray2D]

fit(X: NDArray2D) → None

predict(X: NDArray2D) → Tuple[NDArray2D, NDArray2D]