def test_estimate_transform_spd(self):
point = spd.SPDMatrices(3).random_uniform()
points = gs.stack([point, point])
transformer = ToTangentSpace(geometry=spd.SPDMetricAffine(3))
transformer.fit(X=points)
result = transformer.transform(points)
expected = gs.zeros((2, 6))
self.assertAllClose(expected, result, atol=1e-5)
def test_inverse_transform_spd(self):
point = spd.SPDMatrices(3).random_uniform(10)
transformer = ToTangentSpace(geometry=spd.SPDMetricLogEuclidean(3))
X = transformer.fit_transform(X=point)
result = transformer.inverse_transform(X)
expected = point
self.assertAllClose(expected, result, atol=1e-4)
transformer = ToTangentSpace(geometry=spd.SPDMetricAffine(3))
X = transformer.fit_transform(X=point)
result = transformer.inverse_transform(X)
expected = point
self.assertAllClose(expected, result, atol=1e-4)
def test_spd_kmeans_fit(self):
gs.random.seed(0)
dim = 3
n_points = 2
space = spd_matrices.SPDMatrices(dim)
data = space.random_point(n_samples=n_points)
metric = spd_matrices.SPDMetricAffine(dim)
kmeans = RiemannianKMeans(metric, n_clusters=1, lr=1.0)
kmeans.fit(data)
result = kmeans.centroids
mean = FrechetMean(metric=metric, point_type="matrix", max_iter=100)
mean.fit(data)
expected = mean.estimate_
self.assertAllClose(result, expected)
def __init__(self, n_labels):
super().__init__()
self.n_labels = n_labels
self.ambient = spd.SPDMatrices(n=self.n_labels)
