Skip to content

Instantly share code, notes, and snippets.

@nguigs
nguigs / Grassmann_Kmeans.py
Created December 2, 2022 10:50
Toy example for online kmeans on the Grassmann manifold
"""Riemannian k-means on the Grassmann Manifold.
Nicolas Guigui, 29/11/2022
"""
import geomstats.backend as gs
from geomstats.geometry.grassmannian import Grassmannian
from geomstats.learning.online_kmeans import OnlineKMeans
n_samples = 10