6fa78eebb531215d2fb5f5955a9ff353b6b8ddf7,examples/tangent_pca_so3.py,,main,#,19

Before Change

    data = SO3_GROUP.random_uniform(n_samples=N_SAMPLES)
    mean = METRIC.mean(data)

    eigenvalues, tangent_eigenvecs = METRIC.tangent_pca(data, base_point=mean)
    n_eigenvalues = min(N_SAMPLES, SO3_GROUP.dimension)

    explained_variances = np.cumsum(eigenvalues)

After Change

    ax.set_title("Data in SO3 (black) and Frechet mean (color)")
    visualization.plot(data, ax, space="SO3_GROUP", color="black")
    visualization.plot(mean, ax, space="SO3_GROUP", linewidth=3)
    ax.set_xlim((-2, 2))
    ax.set_ylim((-2, 2))
    ax.set_zlim((-2, 2))
    plt.show()

Instances