Skip to content

Commit

Permalink
Deploying to master from @ geomstats/geomstats@0102cea 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
luisfpereira committed Jan 12, 2024
1 parent 7e476cc commit 935faff
Show file tree
Hide file tree
Showing 193 changed files with 634 additions and 770 deletions.
Binary file modified .doctrees/api/geomstats.geometry.doctree
Binary file not shown.
Binary file modified .doctrees/api/geomstats.test.doctree
Binary file not shown.
Binary file modified .doctrees/environment.pickle
Binary file not shown.
Binary file not shown.
Binary file modified .doctrees/notebooks/01_foundations__manifolds.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file modified .doctrees/notebooks/17_foundations__stratified_spaces.doctree
Binary file not shown.
Binary file not shown.
2 changes: 1 addition & 1 deletion _modules/geomstats/algebra_utils.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
6 changes: 4 additions & 2 deletions _modules/geomstats/datasets/prepare_emg_data.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down Expand Up @@ -550,7 +550,9 @@ <h1>Source code for geomstats.datasets.prepare_emg_data</h1><div class="highligh
<span class="n">covs</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">cov</span><span class="p">(</span><span class="n">x</span><span class="o">.</span><span class="n">transpose</span><span class="p">()))</span>
<span class="bp">self</span><span class="o">.</span><span class="n">labels</span> <span class="o">=</span> <span class="n">gs</span><span class="o">.</span><span class="n">array</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s2">&quot;y&quot;</span><span class="p">][</span><span class="bp">self</span><span class="o">.</span><span class="n">batches</span><span class="p">])</span>
<span class="bp">self</span><span class="o">.</span><span class="n">covs</span> <span class="o">=</span> <span class="n">gs</span><span class="o">.</span><span class="n">array</span><span class="p">(</span><span class="n">covs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">covecs</span> <span class="o">=</span> <span class="n">gs</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="n">SymmetricMatrices</span><span class="o">.</span><span class="n">to_vector</span><span class="p">(</span><span class="n">cov</span><span class="p">)</span> <span class="k">for</span> <span class="n">cov</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">covs</span><span class="p">])</span>
<span class="bp">self</span><span class="o">.</span><span class="n">covecs</span> <span class="o">=</span> <span class="n">gs</span><span class="o">.</span><span class="n">array</span><span class="p">(</span>
<span class="p">[</span><span class="n">SymmetricMatrices</span><span class="o">.</span><span class="n">basis_representation</span><span class="p">(</span><span class="n">cov</span><span class="p">)</span> <span class="k">for</span> <span class="n">cov</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">covs</span><span class="p">]</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">diags</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">covs</span><span class="o">.</span><span class="n">diagonal</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span></div>
</div>

Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/datasets/prepare_graph_data.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/datasets/utils.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/distributions/lognormal.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/errors.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/exceptions.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
111 changes: 87 additions & 24 deletions _modules/geomstats/geometry/base.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down Expand Up @@ -586,14 +586,13 @@ <h1>Source code for geomstats.geometry.base</h1><div class="highlight"><pre>

<span class="sd"> Returns</span>
<span class="sd"> -------</span>
<span class="sd"> point : array-like, shape=[..., *point_shape]</span>
<span class="sd"> point : array-like, shape=[..., dim]</span>
<span class="sd"> Sample.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">size</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">shape</span>
<span class="n">size</span> <span class="o">=</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dim</span><span class="p">,)</span>
<span class="k">if</span> <span class="n">n_samples</span> <span class="o">!=</span> <span class="mi">1</span><span class="p">:</span>
<span class="n">size</span> <span class="o">=</span> <span class="p">(</span><span class="n">n_samples</span><span class="p">,)</span> <span class="o">+</span> <span class="bp">self</span><span class="o">.</span><span class="n">shape</span>
<span class="n">point</span> <span class="o">=</span> <span class="n">bound</span> <span class="o">*</span> <span class="p">(</span><span class="n">gs</span><span class="o">.</span><span class="n">random</span><span class="o">.</span><span class="n">rand</span><span class="p">(</span><span class="o">*</span><span class="n">size</span><span class="p">)</span> <span class="o">-</span> <span class="mf">0.5</span><span class="p">)</span> <span class="o">*</span> <span class="mi">2</span>
<span class="k">return</span> <span class="n">point</span></div>
<span class="n">size</span> <span class="o">=</span> <span class="p">(</span><span class="n">n_samples</span><span class="p">,)</span> <span class="o">+</span> <span class="n">size</span>
<span class="k">return</span> <span class="n">bound</span> <span class="o">*</span> <span class="p">(</span><span class="n">gs</span><span class="o">.</span><span class="n">random</span><span class="o">.</span><span class="n">rand</span><span class="p">(</span><span class="o">*</span><span class="n">size</span><span class="p">)</span> <span class="o">-</span> <span class="mf">0.5</span><span class="p">)</span> <span class="o">*</span> <span class="mi">2</span></div>


<span class="nd">@property</span>
Expand All @@ -609,6 +608,83 @@ <h1>Source code for geomstats.geometry.base</h1><div class="highlight"><pre>



<div class="viewcode-block" id="MatrixVectorSpace">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.MatrixVectorSpace">[docs]</a>
<span class="k">class</span> <span class="nc">MatrixVectorSpace</span><span class="p">(</span><span class="n">VectorSpace</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;A matrix vector space.&quot;&quot;&quot;</span>

<div class="viewcode-block" id="MatrixVectorSpace.basis_representation">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.MatrixVectorSpace.basis_representation">[docs]</a>
<span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
<span class="k">def</span> <span class="nf">basis_representation</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">matrix_representation</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Compute the coefficients of matrices in the given basis.</span>

<span class="sd"> This takes a matrix (the matrix representation of a point) and</span>
<span class="sd"> transforms it into its corresponding vector representation</span>
<span class="sd"> (the coefficients wrt a given basis).</span>

<span class="sd"> Previously, this method was called `to_vector`. `basis_representation`</span>
<span class="sd"> makes it more clear that the vector representation depends on the chosen</span>
<span class="sd"> basis.</span>

<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> matrix_representation : array-like, shape=[..., *point_shape]</span>
<span class="sd"> Matrix.</span>

<span class="sd"> Returns</span>
<span class="sd"> -------</span>
<span class="sd"> basis_representation : array-like, shape=[..., dim]</span>
<span class="sd"> Coefficients in the basis.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">raise</span> <span class="ne">NotImplementedError</span><span class="p">(</span><span class="s2">&quot;basis_representation not implemented.&quot;</span><span class="p">)</span></div>


<div class="viewcode-block" id="MatrixVectorSpace.matrix_representation">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.MatrixVectorSpace.matrix_representation">[docs]</a>
<span class="k">def</span> <span class="nf">matrix_representation</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">basis_representation</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Compute the matrix representation for the given basis coefficients.</span>

<span class="sd"> This takes a vector representation of a point (the coefficients wrt</span>
<span class="sd"> a given basis) and creates the corresponding matrix representation.</span>

<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> basis_representation : array-like, shape=[..., dim]</span>
<span class="sd"> Coefficients in the basis.</span>

<span class="sd"> Returns</span>
<span class="sd"> -------</span>
<span class="sd"> matrix_representation : array-like, shape=[..., *point_shape]</span>
<span class="sd"> Matrix.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">return</span> <span class="n">gs</span><span class="o">.</span><span class="n">einsum</span><span class="p">(</span><span class="s2">&quot;...i,ijk -&gt;...jk&quot;</span><span class="p">,</span> <span class="n">basis_representation</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">basis</span><span class="p">)</span></div>


<div class="viewcode-block" id="MatrixVectorSpace.random_point">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.MatrixVectorSpace.random_point">[docs]</a>
<span class="k">def</span> <span class="nf">random_point</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">n_samples</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span> <span class="n">bound</span><span class="o">=</span><span class="mf">1.0</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Sample in the vector space with a uniform distribution in a box.</span>

<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> n_samples : int</span>
<span class="sd"> Number of samples.</span>
<span class="sd"> Optional, default: 1.</span>
<span class="sd"> bound : float</span>
<span class="sd"> Side of hypercube support of the uniform distribution.</span>
<span class="sd"> Optional, default: 1.0</span>

<span class="sd"> Returns</span>
<span class="sd"> -------</span>
<span class="sd"> point : array-like, shape=[..., *point_shape]</span>
<span class="sd"> Sample.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">matrix_representation</span><span class="p">(</span><span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">random_point</span><span class="p">(</span><span class="n">n_samples</span><span class="p">,</span> <span class="n">bound</span><span class="p">))</span></div>
</div>



<div class="viewcode-block" id="ComplexVectorSpace">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.ComplexVectorSpace">[docs]</a>
<span class="k">class</span> <span class="nc">ComplexVectorSpace</span><span class="p">(</span><span class="n">ComplexManifold</span><span class="p">,</span> <span class="n">abc</span><span class="o">.</span><span class="n">ABC</span><span class="p">):</span>
Expand All @@ -628,7 +704,6 @@ <h1>Source code for geomstats.geometry.base</h1><div class="highlight"><pre>
<span class="k">if</span> <span class="n">dim</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">dim</span> <span class="o">=</span> <span class="n">math</span><span class="o">.</span><span class="n">prod</span><span class="p">(</span><span class="n">shape</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">shape</span><span class="o">=</span><span class="n">shape</span><span class="p">,</span> <span class="n">dim</span><span class="o">=</span><span class="n">dim</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_basis</span> <span class="o">=</span> <span class="kc">None</span>

<div class="viewcode-block" id="ComplexVectorSpace.belongs">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.ComplexVectorSpace.belongs">[docs]</a>
Expand Down Expand Up @@ -758,26 +833,14 @@ <h1>Source code for geomstats.geometry.base</h1><div class="highlight"><pre>
<span class="n">gs</span><span class="o">.</span><span class="n">random</span><span class="o">.</span><span class="n">rand</span><span class="p">(</span><span class="o">*</span><span class="n">size</span><span class="p">,</span> <span class="n">dtype</span><span class="o">=</span><span class="n">gs</span><span class="o">.</span><span class="n">get_default_cdtype</span><span class="p">())</span> <span class="o">-</span> <span class="mf">0.5</span> <span class="o">-</span> <span class="mf">0.5</span><span class="n">j</span>
<span class="p">)</span>
<span class="k">return</span> <span class="n">point</span></div>
</div>


<span class="nd">@property</span>
<span class="k">def</span> <span class="nf">basis</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Basis of the vector space.&quot;&quot;&quot;</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_basis</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_basis</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_create_basis</span><span class="p">()</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_basis</span>

<span class="nd">@basis</span><span class="o">.</span><span class="n">setter</span>
<span class="k">def</span> <span class="nf">basis</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">basis</span><span class="p">):</span>
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">basis</span><span class="p">)</span> <span class="o">&lt;</span> <span class="bp">self</span><span class="o">.</span><span class="n">dim</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span>
<span class="s2">&quot;The basis should have length equal to the dimension of the space.&quot;</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_basis</span> <span class="o">=</span> <span class="n">basis</span>

<span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
<span class="k">def</span> <span class="nf">_create_basis</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Create a canonical basis.&quot;&quot;&quot;</span></div>
<div class="viewcode-block" id="ComplexMatrixVectorSpace">
<a class="viewcode-back" href="../../../api/geomstats.geometry.html#geomstats.geometry.base.ComplexMatrixVectorSpace">[docs]</a>
<span class="k">class</span> <span class="nc">ComplexMatrixVectorSpace</span><span class="p">(</span><span class="n">ComplexVectorSpace</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;A matrix vector space.&quot;&quot;&quot;</span></div>



Expand Down
2 changes: 1 addition & 1 deletion _modules/geomstats/geometry/complex_manifold.html
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
<link rel="search" title="Search" href="../../../search.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docbuild:last-update" content="Jan 9, 2024, 3:28:19 PM"/>
<meta name="docbuild:last-update" content="Jan 12, 2024, 9:08:44 AM"/>
</head>


Expand Down
Loading

0 comments on commit 935faff

Please sign in to comment.