diff --git a/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip b/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip
index 3ed0a4f..71361a8 100644
Binary files a/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip and b/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip differ
diff --git a/dev/_downloads/26989111dccd2d4534f75f59dd77f104/demo_ea_adaptivity.zip b/dev/_downloads/26989111dccd2d4534f75f59dd77f104/demo_ea_adaptivity.zip
index 69d1c64..e32ee72 100644
Binary files a/dev/_downloads/26989111dccd2d4534f75f59dd77f104/demo_ea_adaptivity.zip and b/dev/_downloads/26989111dccd2d4534f75f59dd77f104/demo_ea_adaptivity.zip differ
diff --git a/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip b/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip
index 9afaadb..f4ecfec 100644
Binary files a/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip and b/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip differ
diff --git a/dev/_downloads/9c98a10e6c0743947672bcab8d552167/demo_pca_via_affinity_matcher.zip b/dev/_downloads/9c98a10e6c0743947672bcab8d552167/demo_pca_via_affinity_matcher.zip
index c97ad58..cfc28ed 100644
Binary files a/dev/_downloads/9c98a10e6c0743947672bcab8d552167/demo_pca_via_affinity_matcher.zip and b/dev/_downloads/9c98a10e6c0743947672bcab8d552167/demo_pca_via_affinity_matcher.zip differ
diff --git a/dev/_downloads/a326ed5e0cfd78a82e54fd5203079d4a/demo_tsne_swiss_roll.zip b/dev/_downloads/a326ed5e0cfd78a82e54fd5203079d4a/demo_tsne_swiss_roll.zip
index fee8bd4..855ccf2 100644
Binary files a/dev/_downloads/a326ed5e0cfd78a82e54fd5203079d4a/demo_tsne_swiss_roll.zip and b/dev/_downloads/a326ed5e0cfd78a82e54fd5203079d4a/demo_tsne_swiss_roll.zip differ
diff --git a/dev/_downloads/ccd90bc6ee7cbcb51bd8b71882fb5168/demo_ne_methods_affinity_matcher.zip b/dev/_downloads/ccd90bc6ee7cbcb51bd8b71882fb5168/demo_ne_methods_affinity_matcher.zip
index 5a9eb33..2a371b8 100644
Binary files a/dev/_downloads/ccd90bc6ee7cbcb51bd8b71882fb5168/demo_ne_methods_affinity_matcher.zip and b/dev/_downloads/ccd90bc6ee7cbcb51bd8b71882fb5168/demo_ne_methods_affinity_matcher.zip differ
diff --git a/dev/_sources/auto_examples/affinities/demo_ea_adaptivity.rst.txt b/dev/_sources/auto_examples/affinities/demo_ea_adaptivity.rst.txt
index 08f624a..911080c 100644
--- a/dev/_sources/auto_examples/affinities/demo_ea_adaptivity.rst.txt
+++ b/dev/_sources/auto_examples/affinities/demo_ea_adaptivity.rst.txt
@@ -207,7 +207,7 @@ This affinity is an important component of the **TSNE** algorithm.
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** (0 minutes 2.973 seconds)
+   **Total running time of the script:** (0 minutes 3.166 seconds)
 
 
 .. _sphx_glr_download_auto_examples_affinities_demo_ea_adaptivity.py:
diff --git a/dev/_sources/auto_examples/affinities/sg_execution_times.rst.txt b/dev/_sources/auto_examples/affinities/sg_execution_times.rst.txt
index f45c2b6..0bbbdab 100644
--- a/dev/_sources/auto_examples/affinities/sg_execution_times.rst.txt
+++ b/dev/_sources/auto_examples/affinities/sg_execution_times.rst.txt
@@ -6,7 +6,7 @@
 
 Computation times
 =================
-**00:02.973** total execution time for 1 file **from auto_examples/affinities**:
+**00:03.166** total execution time for 1 file **from auto_examples/affinities**:
 
 .. container::
 
@@ -33,5 +33,5 @@ Computation times
      - Time
      - Mem (MB)
    * - :ref:`sphx_glr_auto_examples_affinities_demo_ea_adaptivity.py` (``demo_ea_adaptivity.py``)
-     - 00:02.973
+     - 00:03.166
      - 0.0
diff --git a/dev/_sources/auto_examples/basics/demo_ne_methods_affinity_matcher.rst.txt b/dev/_sources/auto_examples/basics/demo_ne_methods_affinity_matcher.rst.txt
index bf12ded..268179d 100644
--- a/dev/_sources/auto_examples/basics/demo_ne_methods_affinity_matcher.rst.txt
+++ b/dev/_sources/auto_examples/basics/demo_ne_methods_affinity_matcher.rst.txt
@@ -249,7 +249,7 @@ On the efficiency of using the torchdr API rather than AffinityMatcher directly
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** (0 minutes 56.286 seconds)
+   **Total running time of the script:** (1 minutes 14.869 seconds)
 
 
 .. _sphx_glr_download_auto_examples_basics_demo_ne_methods_affinity_matcher.py:
diff --git a/dev/_sources/auto_examples/basics/demo_pca_via_affinity_matcher.rst.txt b/dev/_sources/auto_examples/basics/demo_pca_via_affinity_matcher.rst.txt
index b35e559..968b7c4 100644
--- a/dev/_sources/auto_examples/basics/demo_pca_via_affinity_matcher.rst.txt
+++ b/dev/_sources/auto_examples/basics/demo_pca_via_affinity_matcher.rst.txt
@@ -158,7 +158,7 @@ both methods.
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** (0 minutes 5.006 seconds)
+   **Total running time of the script:** (0 minutes 5.351 seconds)
 
 
 .. _sphx_glr_download_auto_examples_basics_demo_pca_via_affinity_matcher.py:
diff --git a/dev/_sources/auto_examples/basics/demo_tsne_swiss_roll.rst.txt b/dev/_sources/auto_examples/basics/demo_tsne_swiss_roll.rst.txt
index 22d88c2..5144699 100644
--- a/dev/_sources/auto_examples/basics/demo_tsne_swiss_roll.rst.txt
+++ b/dev/_sources/auto_examples/basics/demo_tsne_swiss_roll.rst.txt
@@ -215,7 +215,7 @@ mistakenly considered neighbors.
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** (1 minutes 7.864 seconds)
+   **Total running time of the script:** (1 minutes 14.668 seconds)
 
 
 .. _sphx_glr_download_auto_examples_basics_demo_tsne_swiss_roll.py:
diff --git a/dev/_sources/auto_examples/basics/sg_execution_times.rst.txt b/dev/_sources/auto_examples/basics/sg_execution_times.rst.txt
index 08eaae1..d7305a0 100644
--- a/dev/_sources/auto_examples/basics/sg_execution_times.rst.txt
+++ b/dev/_sources/auto_examples/basics/sg_execution_times.rst.txt
@@ -6,7 +6,7 @@
 
 Computation times
 =================
-**02:09.157** total execution time for 3 files **from auto_examples/basics**:
+**02:34.888** total execution time for 3 files **from auto_examples/basics**:
 
 .. container::
 
@@ -32,12 +32,12 @@ Computation times
    * - Example
      - Time
      - Mem (MB)
-   * - :ref:`sphx_glr_auto_examples_basics_demo_tsne_swiss_roll.py` (``demo_tsne_swiss_roll.py``)
-     - 01:07.864
-     - 0.0
    * - :ref:`sphx_glr_auto_examples_basics_demo_ne_methods_affinity_matcher.py` (``demo_ne_methods_affinity_matcher.py``)
-     - 00:56.286
+     - 01:14.869
+     - 0.0
+   * - :ref:`sphx_glr_auto_examples_basics_demo_tsne_swiss_roll.py` (``demo_tsne_swiss_roll.py``)
+     - 01:14.668
      - 0.0
    * - :ref:`sphx_glr_auto_examples_basics_demo_pca_via_affinity_matcher.py` (``demo_pca_via_affinity_matcher.py``)
-     - 00:05.006
+     - 00:05.351
      - 0.0
diff --git a/dev/_sources/sg_execution_times.rst.txt b/dev/_sources/sg_execution_times.rst.txt
index 569e145..97cc330 100644
--- a/dev/_sources/sg_execution_times.rst.txt
+++ b/dev/_sources/sg_execution_times.rst.txt
@@ -6,7 +6,7 @@
 
 Computation times
 =================
-**02:12.130** total execution time for 4 files **from all galleries**:
+**02:38.054** total execution time for 4 files **from all galleries**:
 
 .. container::
 
@@ -32,15 +32,15 @@ Computation times
    * - Example
      - Time
      - Mem (MB)
-   * - :ref:`sphx_glr_auto_examples_basics_demo_tsne_swiss_roll.py` (``../../examples/basics/demo_tsne_swiss_roll.py``)
-     - 01:07.864
-     - 0.0
    * - :ref:`sphx_glr_auto_examples_basics_demo_ne_methods_affinity_matcher.py` (``../../examples/basics/demo_ne_methods_affinity_matcher.py``)
-     - 00:56.286
+     - 01:14.869
+     - 0.0
+   * - :ref:`sphx_glr_auto_examples_basics_demo_tsne_swiss_roll.py` (``../../examples/basics/demo_tsne_swiss_roll.py``)
+     - 01:14.668
      - 0.0
    * - :ref:`sphx_glr_auto_examples_basics_demo_pca_via_affinity_matcher.py` (``../../examples/basics/demo_pca_via_affinity_matcher.py``)
-     - 00:05.006
+     - 00:05.351
      - 0.0
    * - :ref:`sphx_glr_auto_examples_affinities_demo_ea_adaptivity.py` (``../../examples/affinities/demo_ea_adaptivity.py``)
-     - 00:02.973
+     - 00:03.166
      - 0.0
diff --git a/dev/auto_examples/affinities/demo_ea_adaptivity.html b/dev/auto_examples/affinities/demo_ea_adaptivity.html
index 4e7ca43..666ef3b 100644
--- a/dev/auto_examples/affinities/demo_ea_adaptivity.html
+++ b/dev/auto_examples/affinities/demo_ea_adaptivity.html
@@ -208,7 +208,7 @@ <h2>Entropic affinity (adaptive bandwidth)<a class="headerlink" href="#entropic-
 <img src="../../_images/sphx_glr_demo_ea_adaptivity_002.png" srcset="../../_images/sphx_glr_demo_ea_adaptivity_002.png" alt="Entropic Affinity Matrix, Entropic Affinity Graph" class = "sphx-glr-single-img"/><p>We can now observe a <strong>homogeneous density of connections</strong> across clusters.
 Thus, the entropic affinity effectively filters out the various noise levels.</p>
 <p>This affinity is an important component of the <strong>TSNE</strong> algorithm.</p>
-<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (0 minutes 2.973 seconds)</p>
+<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (0 minutes 3.166 seconds)</p>
 <div class="sphx-glr-footer sphx-glr-footer-example docutils container" id="sphx-glr-download-auto-examples-affinities-demo-ea-adaptivity-py">
 <div class="sphx-glr-download sphx-glr-download-jupyter docutils container">
 <p><a class="reference download internal" download="" href="../../_downloads/17d429f11ca029172465ca0d7bb1879e/demo_ea_adaptivity.ipynb"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Jupyter</span> <span class="pre">notebook:</span> <span class="pre">demo_ea_adaptivity.ipynb</span></code></a></p>
diff --git a/dev/auto_examples/affinities/sg_execution_times.html b/dev/auto_examples/affinities/sg_execution_times.html
index 17a885c..39a1f46 100644
--- a/dev/auto_examples/affinities/sg_execution_times.html
+++ b/dev/auto_examples/affinities/sg_execution_times.html
@@ -84,7 +84,7 @@
              
   <section id="computation-times">
 <span id="sphx-glr-auto-examples-affinities-sg-execution-times"></span><h1>Computation times<a class="headerlink" href="#computation-times" title="Link to this heading"></a></h1>
-<p><strong>00:02.973</strong> total execution time for 1 file <strong>from auto_examples/affinities</strong>:</p>
+<p><strong>00:03.166</strong> total execution time for 1 file <strong>from auto_examples/affinities</strong>:</p>
 <div class="docutils container">
 <style scoped>
 <link href="https://cdnjs.cloudflare.com/ajax/libs/twitter-bootstrap/5.3.0/css/bootstrap.min.css" rel="stylesheet" />
@@ -106,7 +106,7 @@
 </thead>
 <tbody>
 <tr class="row-even"><td><p><a class="reference internal" href="demo_ea_adaptivity.html#sphx-glr-auto-examples-affinities-demo-ea-adaptivity-py"><span class="std std-ref">Entropic Affinities can adapt to varying noise levels</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_ea_adaptivity.py</span></code>)</p></td>
-<td><p>00:02.973</p></td>
+<td><p>00:03.166</p></td>
 <td><p>0.0</p></td>
 </tr>
 </tbody>
diff --git a/dev/auto_examples/basics/demo_ne_methods_affinity_matcher.html b/dev/auto_examples/basics/demo_ne_methods_affinity_matcher.html
index 71c3688..106cd79 100644
--- a/dev/auto_examples/basics/demo_ne_methods_affinity_matcher.html
+++ b/dev/auto_examples/basics/demo_ne_methods_affinity_matcher.html
@@ -240,7 +240,7 @@ <h2>On the efficiency of using the torchdr API rather than AffinityMatcher direc
 In TorchDR, it is therefore recommended to use the specific class associated
 with the desired algorithm when available.</p>
 </div>
-<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (0 minutes 56.286 seconds)</p>
+<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (1 minutes 14.869 seconds)</p>
 <div class="sphx-glr-footer sphx-glr-footer-example docutils container" id="sphx-glr-download-auto-examples-basics-demo-ne-methods-affinity-matcher-py">
 <div class="sphx-glr-download sphx-glr-download-jupyter docutils container">
 <p><a class="reference download internal" download="" href="../../_downloads/286dcc8a82b9a5553a5d809cc0f6fa61/demo_ne_methods_affinity_matcher.ipynb"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Jupyter</span> <span class="pre">notebook:</span> <span class="pre">demo_ne_methods_affinity_matcher.ipynb</span></code></a></p>
diff --git a/dev/auto_examples/basics/demo_pca_via_affinity_matcher.html b/dev/auto_examples/basics/demo_pca_via_affinity_matcher.html
index 381f22c..c2e4892 100644
--- a/dev/auto_examples/basics/demo_pca_via_affinity_matcher.html
+++ b/dev/auto_examples/basics/demo_pca_via_affinity_matcher.html
@@ -170,7 +170,7 @@ <h2>PCA via AffinityMatcher<a class="headerlink" href="#pca-via-affinitymatcher"
 </div>
 <img src="../../_images/sphx_glr_demo_pca_via_affinity_matcher_002.png" srcset="../../_images/sphx_glr_demo_pca_via_affinity_matcher_002.png" alt="PCA via AffinityMatcher" class = "sphx-glr-single-img"/><p>We can see that we obtain the same PCA embedding (up to a rotation) using
 both methods.</p>
-<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (0 minutes 5.006 seconds)</p>
+<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (0 minutes 5.351 seconds)</p>
 <div class="sphx-glr-footer sphx-glr-footer-example docutils container" id="sphx-glr-download-auto-examples-basics-demo-pca-via-affinity-matcher-py">
 <div class="sphx-glr-download sphx-glr-download-jupyter docutils container">
 <p><a class="reference download internal" download="" href="../../_downloads/01119b09b904483a657e921d21bc15d1/demo_pca_via_affinity_matcher.ipynb"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Jupyter</span> <span class="pre">notebook:</span> <span class="pre">demo_pca_via_affinity_matcher.ipynb</span></code></a></p>
diff --git a/dev/auto_examples/basics/demo_tsne_swiss_roll.html b/dev/auto_examples/basics/demo_tsne_swiss_roll.html
index 9f68a35..d1a79f4 100644
--- a/dev/auto_examples/basics/demo_tsne_swiss_roll.html
+++ b/dev/auto_examples/basics/demo_tsne_swiss_roll.html
@@ -186,7 +186,7 @@ <h2>See the impact of perplexity<a class="headerlink" href="#see-the-impact-of-p
 and fails to capture the manifold’s geometry. Conversely, when the perplexity is too
 high, points that are distant on the manifold but close in the ambient space are
 mistakenly considered neighbors.</p>
-<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (1 minutes 7.864 seconds)</p>
+<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> (1 minutes 14.668 seconds)</p>
 <div class="sphx-glr-footer sphx-glr-footer-example docutils container" id="sphx-glr-download-auto-examples-basics-demo-tsne-swiss-roll-py">
 <div class="sphx-glr-download sphx-glr-download-jupyter docutils container">
 <p><a class="reference download internal" download="" href="../../_downloads/d07a5a618e87545c420604821fa60541/demo_tsne_swiss_roll.ipynb"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Jupyter</span> <span class="pre">notebook:</span> <span class="pre">demo_tsne_swiss_roll.ipynb</span></code></a></p>
diff --git a/dev/auto_examples/basics/sg_execution_times.html b/dev/auto_examples/basics/sg_execution_times.html
index 5dcf3f3..c2cd785 100644
--- a/dev/auto_examples/basics/sg_execution_times.html
+++ b/dev/auto_examples/basics/sg_execution_times.html
@@ -84,7 +84,7 @@
              
   <section id="computation-times">
 <span id="sphx-glr-auto-examples-basics-sg-execution-times"></span><h1>Computation times<a class="headerlink" href="#computation-times" title="Link to this heading"></a></h1>
-<p><strong>02:09.157</strong> total execution time for 3 files <strong>from auto_examples/basics</strong>:</p>
+<p><strong>02:34.888</strong> total execution time for 3 files <strong>from auto_examples/basics</strong>:</p>
 <div class="docutils container">
 <style scoped>
 <link href="https://cdnjs.cloudflare.com/ajax/libs/twitter-bootstrap/5.3.0/css/bootstrap.min.css" rel="stylesheet" />
@@ -105,16 +105,16 @@
 </tr>
 </thead>
 <tbody>
-<tr class="row-even"><td><p><a class="reference internal" href="demo_tsne_swiss_roll.html#sphx-glr-auto-examples-basics-demo-tsne-swiss-roll-py"><span class="std std-ref">TSNE embedding of the swiss roll dataset</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_tsne_swiss_roll.py</span></code>)</p></td>
-<td><p>01:07.864</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="demo_ne_methods_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-ne-methods-affinity-matcher-py"><span class="std std-ref">Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_ne_methods_affinity_matcher.py</span></code>)</p></td>
+<td><p>01:14.869</p></td>
 <td><p>0.0</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="demo_ne_methods_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-ne-methods-affinity-matcher-py"><span class="std std-ref">Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_ne_methods_affinity_matcher.py</span></code>)</p></td>
-<td><p>00:56.286</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="demo_tsne_swiss_roll.html#sphx-glr-auto-examples-basics-demo-tsne-swiss-roll-py"><span class="std std-ref">TSNE embedding of the swiss roll dataset</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_tsne_swiss_roll.py</span></code>)</p></td>
+<td><p>01:14.668</p></td>
 <td><p>0.0</p></td>
 </tr>
 <tr class="row-even"><td><p><a class="reference internal" href="demo_pca_via_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-pca-via-affinity-matcher-py"><span class="std std-ref">PCA via SVD and via AffinityMatcher</span></a> (<code class="docutils literal notranslate"><span class="pre">demo_pca_via_affinity_matcher.py</span></code>)</p></td>
-<td><p>00:05.006</p></td>
+<td><p>00:05.351</p></td>
 <td><p>0.0</p></td>
 </tr>
 </tbody>
diff --git a/dev/searchindex.js b/dev/searchindex.js
index 49091fb..19c25af 100644
--- a/dev/searchindex.js
+++ b/dev/searchindex.js
@@ -1 +1 @@
-Search.setIndex({"alltitles": {"API and Modules": [[0, null]], "Advanced Dimensionality Reduction with TorchDR": [[0, "advanced-dimensionality-reduction-with-torchdr"]], "Affinities": [[7, "affinities"], [39, "affinities"], [49, "affinities"]], "Affinities Normalized by kNN Distances": [[0, "affinities-normalized-by-knn-distances"]], "Affinity": [[41, null]], "Affinity Classes": [[0, "affinity-classes"]], "AffinityMatcher": [[9, null], [42, null]], "Avoiding memory overflows with KeOps symbolic (lazy) tensors": [[49, "avoiding-memory-overflows-with-keops-symbolic-lazy-tensors"]], "Base Classes": [[0, "base-classes"]], "Base Neighbor Embedding Modules": [[0, "base-neighbor-embedding-modules"]], "Base structure": [[49, "base-structure"]], "Basics": [[7, "basics"]], "Bibliography": [[45, null]], "Citation": [[39, "citation"]], "Computation times": [[2, null], [6, null], [8, null], [40, null]], "Compute the TSNE embedding": [[5, "compute-the-tsne-embedding"]], "DRModule": [[10, null], [43, null]], "Dimensionality Reduction Algorithms": [[39, "dimensionality-reduction-algorithms"]], "Dimensionality Reduction Modules": [[49, "dimensionality-reduction-modules"]], "Dimensionality Reduction sklearn Compatible Estimators": [[0, "dimensionality-reduction-sklearn-compatible-estimators"]], "DoublyStochasticQuadraticAffinity": [[11, null]], "Entropic Affinities": [[0, "entropic-affinities"]], "Entropic Affinities can adapt to varying noise levels": [[1, null]], "Entropic affinity (adaptive bandwidth)": [[1, "entropic-affinity-adaptive-bandwidth"]], "EntropicAffinity": [[12, null]], "Evaluation Metric": [[39, "evaluation-metric"]], "Examples using Affinity:": [[41, "examples-using-affinity"]], "Examples using AffinityMatcher:": [[9, "examples-using-affinitymatcher"], [42, "examples-using-affinitymatcher"], [47, "examples-using-affinitymatcher"]], "Examples using DRModule:": [[10, "examples-using-drmodule"], [43, "examples-using-drmodule"]], "Examples using EntropicAffinity:": [[12, "examples-using-entropicaffinity"], [49, "examples-using-entropicaffinity"]], "Examples using LogAffinity:": [[44, "examples-using-logaffinity"]], "Examples using NeighborEmbedding:": [[19, "examples-using-neighborembedding"]], "Examples using NormalizedGaussianAffinity:": [[20, "examples-using-normalizedgaussianaffinity"]], "Examples using PCA:": [[22, "examples-using-pca"]], "Examples using SNE:": [[23, "examples-using-sne"]], "Examples using SampledNeighborEmbedding:": [[24, "examples-using-sampledneighborembedding"]], "Examples using ScalarProductAffinity:": [[25, "examples-using-scalarproductaffinity"]], "Examples using SparseNeighborEmbedding:": [[28, "examples-using-sparseneighborembedding"]], "Examples using TSNE:": [[31, "examples-using-tsne"]], "Examples using UMAP:": [[33, "examples-using-umap"]], "Finding Help": [[39, "finding-help"]], "Gallery": [[7, null]], "GaussianAffinity": [[13, null]], "General Formulation of Dimensionality Reduction": [[49, "general-formulation-of-dimensionality-reduction"]], "Getting Started": [[39, "getting-started"]], "How to Contribute": [[46, null]], "Implemented Features (to date)": [[39, "implemented-features-to-date"]], "IncrementalPCA": [[14, null]], "InfoTSNE": [[15, null]], "Installation": [[39, "installation"], [47, "installation"]], "KernelPCA": [[16, null]], "LargeVis": [[17, null]], "Load the SNARE-seq dataset (gene expression) with cell type labels": [[3, "load-the-snare-seq-dataset-gene-expression-with-cell-type-labels"]], "Load toy images": [[4, "load-toy-images"], [5, "load-toy-images"]], "LogAffinity": [[44, null]], "MAGICAffinity": [[18, null]], "Neighbor Embedding": [[0, "neighbor-embedding"], [49, "neighbor-embedding"]], "Neighbor Embedding on genomics & equivalent affinity matcher formulation": [[3, null]], "NeighborEmbedding": [[19, null]], "New contributor tips": [[46, "new-contributor-tips"]], "NormalizedGaussianAffinity": [[20, null]], "NormalizedStudentAffinity": [[21, null]], "On the efficiency of using the torchdr API rather than AffinityMatcher directly": [[3, "on-the-efficiency-of-using-the-torchdr-api-rather-than-affinitymatcher-directly"]], "Other various affinities": [[49, "other-various-affinities"]], "Overview": [[49, "overview"]], "PCA": [[22, null]], "PCA via AffinityMatcher": [[4, "pca-via-affinitymatcher"]], "PCA via SVD": [[4, "pca-via-svd"]], "PCA via SVD and via AffinityMatcher": [[4, null]], "Plot the TSNE embedding": [[5, "plot-the-tsne-embedding"]], "Plot the dataset": [[5, "plot-the-dataset"]], "Plot the different embeddings": [[3, "plot-the-different-embeddings"]], "Quadratic Affinities": [[0, "quadratic-affinities"]], "Quick Start Guide": [[47, null]], "Releases": [[48, null]], "Row-normalised Gaussian affinity with constant bandwidth": [[1, "row-normalised-gaussian-affinity-with-constant-bandwidth"]], "Run neighbor embedding methods": [[3, "run-neighbor-embedding-methods"]], "SNE": [[23, null]], "SampledNeighborEmbedding": [[24, null]], "ScalarProductAffinity": [[25, null]], "See the impact of perplexity": [[5, "see-the-impact-of-perplexity"]], "SelfTuningAffinity": [[26, null]], "Simple Affinities": [[0, "simple-affinities"]], "SinkhornAffinity": [[27, null]], "SparseNeighborEmbedding": [[28, null]], "Spectral Embedding": [[0, "spectral-embedding"]], "Spectral methods": [[49, "spectral-methods"]], "Spotlight on affinities based on entropic projections": [[49, "spotlight-on-affinities-based-on-entropic-projections"]], "StudentAffinity": [[29, null]], "SymmetricEntropicAffinity": [[30, null]], "TSNE": [[31, null]], "TSNE embedding of the swiss roll dataset": [[5, null]], "TSNEkhorn": [[32, null]], "Table of Contents": [[49, "table-of-contents"]], "Torch Dimensionality Reduction": [[39, null]], "Torch GPU support and automatic differentiation": [[49, "torch-gpu-support-and-automatic-differentiation"]], "UMAP": [[33, null]], "UMAP Affinities": [[0, "umap-affinities"]], "UMAPAffinityIn": [[34, null]], "UMAPAffinityOut": [[35, null]], "User Guide": [[49, null]], "Using AffinityMatcher": [[3, "using-affinitymatcher"]], "Utils": [[0, "utils"]], "Version 0.1 (2024-09-17)": [[48, "version-0-1-2024-09-17"]], "When to use TorchDR": [[47, "when-to-use-torchdr"]], "binary_search": [[36, null]], "false_position": [[37, null]], "pairwise_distances": [[38, null]]}, "docnames": ["all", "auto_examples/affinities/demo_ea_adaptivity", "auto_examples/affinities/sg_execution_times", "auto_examples/basics/demo_ne_methods_affinity_matcher", "auto_examples/basics/demo_pca_via_affinity_matcher", "auto_examples/basics/demo_tsne_swiss_roll", "auto_examples/basics/sg_execution_times", "auto_examples/index", "auto_examples/sg_execution_times", "gen_modules/torchdr.AffinityMatcher", "gen_modules/torchdr.DRModule", "gen_modules/torchdr.DoublyStochasticQuadraticAffinity", "gen_modules/torchdr.EntropicAffinity", "gen_modules/torchdr.GaussianAffinity", "gen_modules/torchdr.IncrementalPCA", "gen_modules/torchdr.InfoTSNE", "gen_modules/torchdr.KernelPCA", "gen_modules/torchdr.LargeVis", "gen_modules/torchdr.MAGICAffinity", "gen_modules/torchdr.NeighborEmbedding", "gen_modules/torchdr.NormalizedGaussianAffinity", "gen_modules/torchdr.NormalizedStudentAffinity", "gen_modules/torchdr.PCA", "gen_modules/torchdr.SNE", "gen_modules/torchdr.SampledNeighborEmbedding", "gen_modules/torchdr.ScalarProductAffinity", "gen_modules/torchdr.SelfTuningAffinity", "gen_modules/torchdr.SinkhornAffinity", "gen_modules/torchdr.SparseNeighborEmbedding", "gen_modules/torchdr.StudentAffinity", "gen_modules/torchdr.SymmetricEntropicAffinity", "gen_modules/torchdr.TSNE", "gen_modules/torchdr.TSNEkhorn", "gen_modules/torchdr.UMAP", "gen_modules/torchdr.UMAPAffinityIn", "gen_modules/torchdr.UMAPAffinityOut", "gen_modules/torchdr.binary_search", "gen_modules/torchdr.false_position", "gen_modules/torchdr.pairwise_distances", "index", "sg_execution_times", "stubs/torchdr.Affinity", "stubs/torchdr.AffinityMatcher", "stubs/torchdr.DRModule", "stubs/torchdr.LogAffinity", "torchdr.bibliography", "torchdr.contributing", "torchdr.quick_start", "torchdr.releases", "torchdr.user_guide"], "envversion": {"sphinx": 64, "sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx.ext.viewcode": 1, "sphinxcontrib.bibtex": 9}, "filenames": ["all.rst", "auto_examples/affinities/demo_ea_adaptivity.rst", "auto_examples/affinities/sg_execution_times.rst", "auto_examples/basics/demo_ne_methods_affinity_matcher.rst", "auto_examples/basics/demo_pca_via_affinity_matcher.rst", "auto_examples/basics/demo_tsne_swiss_roll.rst", "auto_examples/basics/sg_execution_times.rst", "auto_examples/index.rst", "auto_examples/sg_execution_times.rst", "gen_modules/torchdr.AffinityMatcher.rst", "gen_modules/torchdr.DRModule.rst", "gen_modules/torchdr.DoublyStochasticQuadraticAffinity.rst", "gen_modules/torchdr.EntropicAffinity.rst", "gen_modules/torchdr.GaussianAffinity.rst", "gen_modules/torchdr.IncrementalPCA.rst", "gen_modules/torchdr.InfoTSNE.rst", "gen_modules/torchdr.KernelPCA.rst", "gen_modules/torchdr.LargeVis.rst", "gen_modules/torchdr.MAGICAffinity.rst", "gen_modules/torchdr.NeighborEmbedding.rst", "gen_modules/torchdr.NormalizedGaussianAffinity.rst", "gen_modules/torchdr.NormalizedStudentAffinity.rst", "gen_modules/torchdr.PCA.rst", "gen_modules/torchdr.SNE.rst", "gen_modules/torchdr.SampledNeighborEmbedding.rst", "gen_modules/torchdr.ScalarProductAffinity.rst", "gen_modules/torchdr.SelfTuningAffinity.rst", "gen_modules/torchdr.SinkhornAffinity.rst", "gen_modules/torchdr.SparseNeighborEmbedding.rst", "gen_modules/torchdr.StudentAffinity.rst", "gen_modules/torchdr.SymmetricEntropicAffinity.rst", "gen_modules/torchdr.TSNE.rst", "gen_modules/torchdr.TSNEkhorn.rst", "gen_modules/torchdr.UMAP.rst", "gen_modules/torchdr.UMAPAffinityIn.rst", "gen_modules/torchdr.UMAPAffinityOut.rst", "gen_modules/torchdr.binary_search.rst", "gen_modules/torchdr.false_position.rst", "gen_modules/torchdr.pairwise_distances.rst", "index.rst", "sg_execution_times.rst", "stubs/torchdr.Affinity.rst", "stubs/torchdr.AffinityMatcher.rst", "stubs/torchdr.DRModule.rst", "stubs/torchdr.LogAffinity.rst", "torchdr.bibliography.rst", "torchdr.contributing.rst", "torchdr.quick_start.rst", "torchdr.releases.rst", "torchdr.user_guide.rst"], "indexentries": {"affinity (class in torchdr)": [[41, "torchdr.Affinity", false]], "affinitymatcher (class in torchdr)": [[9, "torchdr.AffinityMatcher", false], [42, "torchdr.AffinityMatcher", false]], "binary_search() (in module torchdr)": [[36, "torchdr.binary_search", false]], "doublystochasticquadraticaffinity (class in torchdr)": [[11, "torchdr.DoublyStochasticQuadraticAffinity", false]], "drmodule (class in torchdr)": [[10, "torchdr.DRModule", false], [43, "torchdr.DRModule", false]], "entropicaffinity (class in torchdr)": [[12, "torchdr.EntropicAffinity", false]], "false_position() (in module torchdr)": [[37, "torchdr.false_position", false]], "fit() (torchdr.affinitymatcher method)": [[9, "torchdr.AffinityMatcher.fit", false], [42, "torchdr.AffinityMatcher.fit", false]], "fit() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.fit", false]], "fit() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.fit", false]], "fit() (torchdr.pca method)": [[22, "torchdr.PCA.fit", false]], "fit_transform() (torchdr.affinitymatcher method)": [[9, "torchdr.AffinityMatcher.fit_transform", false], [42, "torchdr.AffinityMatcher.fit_transform", false]], "fit_transform() (torchdr.drmodule method)": [[10, "torchdr.DRModule.fit_transform", false], [43, "torchdr.DRModule.fit_transform", false]], "fit_transform() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.fit_transform", false]], "fit_transform() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.fit_transform", false]], "fit_transform() (torchdr.pca method)": [[22, "torchdr.PCA.fit_transform", false]], "gaussianaffinity (class in torchdr)": [[13, "torchdr.GaussianAffinity", false]], "gen_batches() (torchdr.incrementalpca static method)": [[14, "torchdr.IncrementalPCA.gen_batches", false]], "incrementalpca (class in torchdr)": [[14, "torchdr.IncrementalPCA", false]], "infotsne (class in torchdr)": [[15, "torchdr.InfoTSNE", false]], "kernelpca (class in torchdr)": [[16, "torchdr.KernelPCA", false]], "largevis (class in torchdr)": [[17, "torchdr.LargeVis", false]], "logaffinity (class in torchdr)": [[44, "torchdr.LogAffinity", false]], "magicaffinity (class in torchdr)": [[18, "torchdr.MAGICAffinity", false]], "module": [[0, "module-torchdr", false], [47, "module-torchdr", false], [49, "module-torchdr", false]], "neighborembedding (class in torchdr)": [[19, "torchdr.NeighborEmbedding", false]], "normalizedgaussianaffinity (class in torchdr)": [[20, "torchdr.NormalizedGaussianAffinity", false]], "normalizedstudentaffinity (class in torchdr)": [[21, "torchdr.NormalizedStudentAffinity", false]], "pairwise_distances() (in module torchdr)": [[38, "torchdr.pairwise_distances", false]], "partial_fit() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.partial_fit", false]], "pca (class in torchdr)": [[22, "torchdr.PCA", false]], "sampledneighborembedding (class in torchdr)": [[24, "torchdr.SampledNeighborEmbedding", false]], "scalarproductaffinity (class in torchdr)": [[25, "torchdr.ScalarProductAffinity", false]], "selftuningaffinity (class in torchdr)": [[26, "torchdr.SelfTuningAffinity", false]], "set_fit_request() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.set_fit_request", false]], "set_partial_fit_request() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.set_partial_fit_request", false]], "sinkhornaffinity (class in torchdr)": [[27, "torchdr.SinkhornAffinity", false]], "sne (class in torchdr)": [[23, "torchdr.SNE", false]], "sparseneighborembedding (class in torchdr)": [[28, "torchdr.SparseNeighborEmbedding", false]], "studentaffinity (class in torchdr)": [[29, "torchdr.StudentAffinity", false]], "symmetricentropicaffinity (class in torchdr)": [[30, "torchdr.SymmetricEntropicAffinity", false]], "torchdr": [[0, "module-torchdr", false], [47, "module-torchdr", false], [49, "module-torchdr", false]], "transform() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.transform", false]], "transform() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.transform", false]], "transform() (torchdr.pca method)": [[22, "torchdr.PCA.transform", false]], "tsne (class in torchdr)": [[31, "torchdr.TSNE", false]], "tsnekhorn (class in torchdr)": [[32, "torchdr.TSNEkhorn", false]], "umap (class in torchdr)": [[33, "torchdr.UMAP", false]], "umapaffinityin (class in torchdr)": [[34, "torchdr.UMAPAffinityIn", false]], "umapaffinityout (class in torchdr)": [[35, "torchdr.UMAPAffinityOut", false]]}, "objects": {"": [[49, 0, 0, "-", "torchdr"]], "torchdr": [[41, 1, 1, "", "Affinity"], [42, 1, 1, "", "AffinityMatcher"], [43, 1, 1, "", "DRModule"], [11, 1, 1, "", "DoublyStochasticQuadraticAffinity"], [12, 1, 1, "", "EntropicAffinity"], [13, 1, 1, "", "GaussianAffinity"], [14, 1, 1, "", "IncrementalPCA"], [15, 1, 1, "", "InfoTSNE"], [16, 1, 1, "", "KernelPCA"], [17, 1, 1, "", "LargeVis"], [44, 1, 1, "", "LogAffinity"], [18, 1, 1, "", "MAGICAffinity"], [19, 1, 1, "", "NeighborEmbedding"], [20, 1, 1, "", "NormalizedGaussianAffinity"], [21, 1, 1, "", "NormalizedStudentAffinity"], [22, 1, 1, "", "PCA"], [23, 1, 1, "", "SNE"], [24, 1, 1, "", "SampledNeighborEmbedding"], [25, 1, 1, "", "ScalarProductAffinity"], [26, 1, 1, "", "SelfTuningAffinity"], [27, 1, 1, "", "SinkhornAffinity"], [28, 1, 1, "", "SparseNeighborEmbedding"], [29, 1, 1, "", "StudentAffinity"], [30, 1, 1, "", "SymmetricEntropicAffinity"], [31, 1, 1, "", "TSNE"], [32, 1, 1, "", "TSNEkhorn"], [33, 1, 1, "", "UMAP"], [34, 1, 1, "", "UMAPAffinityIn"], [35, 1, 1, "", "UMAPAffinityOut"], [36, 3, 1, "", "binary_search"], [37, 3, 1, "", "false_position"], [38, 3, 1, "", "pairwise_distances"]], "torchdr.AffinityMatcher": [[42, 2, 1, "", "fit"], [42, 2, 1, "", "fit_transform"]], "torchdr.DRModule": [[43, 2, 1, "", "fit_transform"]], "torchdr.IncrementalPCA": [[14, 2, 1, "", "fit"], [14, 2, 1, "", "fit_transform"], [14, 2, 1, "", "gen_batches"], [14, 2, 1, "", "partial_fit"], [14, 2, 1, "", "set_fit_request"], [14, 2, 1, "", "set_partial_fit_request"], [14, 2, 1, "", "transform"]], "torchdr.KernelPCA": [[16, 2, 1, "", "fit"], [16, 2, 1, "", "fit_transform"], [16, 2, 1, "", "transform"]], "torchdr.PCA": [[22, 2, 1, "", "fit"], [22, 2, 1, "", "fit_transform"], [22, 2, 1, "", "transform"]]}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "function", "Python function"]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:function"}, "terms": {"": [3, 4, 5, 7, 9, 12, 14, 30, 33, 42, 45], "0": [1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 40, 42, 43], "00": [2, 6, 8, 40], "000": 8, "0001": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "001": [12, 15, 17, 23, 30, 31, 32, 33], "006": [4, 6, 40], "01": [6, 40], "01816": 45, "02": [2, 6, 40], "03426": [33, 45], "05": [5, 6, 11, 27, 34, 40], "07": [6, 9, 15, 17, 19, 23, 24, 28, 31, 33, 40, 42], "09": [6, 36, 37], "09816": 45, "1": [1, 2, 3, 4, 5, 9, 11, 12, 13, 14, 15, 17, 19, 20, 21, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 35, 42, 45, 46, 49], "10": [1, 3, 5, 23, 32, 33], "100": [3, 15, 17, 23, 31, 32, 33, 49], "1000": [9, 11, 12, 27, 34, 36, 37, 42], "11": 45, "111": 5, "12": [3, 5, 15, 17, 31, 40, 45], "125": 48, "127": 48, "130": [40, 48], "137": 48, "143": 48, "147": 48, "15": 3, "157": 6, "174": 45, "18": 45, "1802": [33, 45], "19": 45, "1967": [27, 45], "1d": 36, "1e": [9, 11, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "1e0": [3, 9, 19, 24, 28, 32, 42], "1e1": 4, "2": [1, 3, 4, 5, 9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 21, 22, 23, 24, 28, 29, 30, 31, 32, 33, 42, 43, 45, 46, 49], "20": [1, 5], "200": 5, "2000": [15, 19, 23, 24, 28, 31, 32, 33], "2002": [12, 23, 45], "2004": [26, 45, 49], "2008": [12, 31, 45], "2011": [45, 49], "2013": [27, 45], "2016": [17, 45], "2018": [18, 33, 34, 35, 45, 49], "2019": [27, 45, 49], "2021": [33, 45, 49], "2022": [15, 45], "2023": [11, 45, 49], "2024": [12, 30, 32, 39, 45, 49], "21": 45, "22": 45, "2206": 45, "22nd": 45, "2307": 45, "24": 3, "25": 27, "250": [15, 17, 23, 31, 32, 33], "25th": 45, "2681": 45, "2690": 45, "2825": 45, "2830": 45, "286": [3, 6, 40], "287": 45, "297": 45, "3": [1, 3, 4, 5, 6, 12, 14, 45, 46], "30": [12, 15, 17, 23, 30, 31, 32, 33, 34, 39], "3000": 17, "34": [33, 45], "343": 45, "348": 45, "3d": 5, "4": [3, 5, 9, 14, 15, 17, 19, 24, 28, 31, 32, 33, 40, 42], "47": 45, "5": [1, 4, 5, 14, 15, 17, 24, 30], "50": [4, 5, 15, 39], "500": [5, 30], "56": [3, 6, 40], "5798": 45, "5809": 45, "6": [1, 5, 45], "66": 5, "7": [5, 9, 15, 17, 18, 19, 24, 26, 28, 31, 33, 42], "716": 45, "729": 45, "74": 45, "8": [3, 5], "864": [5, 6, 40], "973": [1, 2, 40], "A": [8, 12, 14, 24, 39, 45, 46, 48, 49], "As": 1, "By": [1, 15, 17, 23, 31, 32, 33], "For": [0, 14, 47, 49], "If": [9, 12, 14, 19, 24, 27, 28, 32, 38, 39, 41, 42, 44, 46, 49], "In": [0, 1, 3, 4, 12, 18, 24, 26, 28, 45, 49], "It": [0, 9, 12, 14, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 39, 42, 49], "Its": [13, 21, 25, 29, 35, 39], "On": [33, 45], "Such": 1, "The": [0, 1, 3, 9, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 30, 31, 32, 33, 34, 36, 37, 38, 39, 41, 42, 44, 45, 46, 49], "Then": 18, "These": [0, 49], "To": [1, 3, 39, 47, 49], "With": 12, "_": [5, 11, 12, 19, 24, 27, 28, 30, 36, 37, 49], "_1": 49, "_2": 11, "_d": 49, "_i": [18, 26, 49], "_j": 49, "_n": 49, "_type_": [15, 17, 31, 32], "a_x": [9, 42, 49], "a_z": [9, 42, 49], "abc": [10, 41, 43], "abil": 0, "abov": [18, 26, 32, 49], "abstract": [10, 43], "acceler": [0, 14, 39, 47, 49], "accept": 39, "accord": [20, 21], "account": 46, "acknowledg": 46, "across": [1, 12, 47], "act": 12, "activ": 46, "ad": [7, 14, 30, 48], "adam": [3, 9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42, 45], "adapt": [2, 7, 12, 14, 20, 39, 40, 41, 44, 48, 49], "add": 0, "add_ax": 5, "add_subplot": [3, 5], "addit": [9, 19, 24, 28, 41, 42, 44], "addition": 49, "address": 46, "adequ": 14, "advanc": [33, 45], "aff": 1, "aff_ea": 1, "affin": [2, 4, 6, 9, 10, 11, 12, 13, 15, 16, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 40, 42, 43, 44, 45, 47, 48], "affinity_in": [3, 4, 9, 19, 24, 28, 42, 49], "affinity_matrix": 49, "affinity_out": [3, 4, 9, 19, 24, 28, 42, 49], "affinitymatch": [0, 6, 7, 10, 19, 22, 25, 40, 41, 43, 48], "after": [14, 19, 46], "aim": [39, 49], "akin": 12, "al": [11, 12, 15, 17, 18, 27, 30, 32, 33, 34, 35, 49], "alain": 45, "alexandr": 45, "alexi": 45, "algorithm": [1, 3, 11, 12, 15, 17, 20, 23, 27, 30, 31, 32, 33, 34, 36, 37, 49], "alia": 14, "all": [0, 7, 11, 12, 27, 30, 39, 40, 46, 47, 48, 49], "all_method": 3, "allow": [0, 14, 30, 35, 46, 47, 49], "along": [18, 20, 21, 26], "alpha": [1, 5], "also": [12, 46], "amari": 45, "ambient": 5, "ambros": 45, "among": 12, "amount": [9, 19, 24, 28, 30, 42], "an": [1, 3, 14, 24, 28, 36, 37, 39, 45, 46, 48, 49], "analog": 49, "analysi": [14, 16, 22, 49], "ani": [39, 46, 48, 49], "anoth": [11, 27, 49], "antiga": 45, "apart": 35, "api": [39, 46, 49], "append": 5, "appli": 14, "applic": 48, "approach": [4, 47, 49], "appropri": 46, "approv": 46, "approxim": [0, 14, 33, 45, 49], "ar": [0, 5, 10, 12, 14, 16, 22, 27, 35, 39, 43, 46, 47, 49], "arg": [11, 12, 27, 30], "argument": [9, 14, 15, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42], "around": [1, 49], "arrai": [3, 39], "artifici": 45, "arxiv": [33, 45], "ascent": [11, 27, 30], "aspect": 49, "assel": [1, 3, 4, 5, 12, 30, 32, 39, 45, 49], "assist": 46, "associ": [3, 39, 49], "astyp": 39, "attract": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "aur\u00e9lien": 39, "author": [1, 3, 4, 5, 39], "authorship": 46, "auto": [9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44], "auto_exampl": [2, 6, 8], "auto_examples_jupyt": 7, "auto_examples_python": 7, "autodiff": 45, "automat": [9, 42, 46], "avail": [1, 3, 4, 12, 14, 22, 39, 49], "averag": [12, 18], "avoid": 39, "awai": 49, "ax": [3, 5], "azim": 5, "b": [33, 35, 45], "back": 46, "backend": [39, 48], "balanc": 49, "bandwidth": [13, 18, 20, 26, 49], "base": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 39, 41, 42, 43, 44, 48], "base_kernel": [11, 27, 49], "baseestim": [10, 43, 49], "basic": [3, 6, 40, 48, 49], "batch": [14, 36, 37, 39], "batch_siz": 14, "becaus": 1, "been": 49, "befor": 46, "begin": [36, 37], "benchmark": 47, "benefit": [39, 49], "benjamin": 45, "bernhard": 45, "bertrand": 45, "best": [39, 49], "between": [0, 3, 11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 38, 45, 49], "bibtex": [39, 48], "big": [12, 15, 23, 31], "binari": [36, 49], "black": [1, 46], "block": [39, 49], "blog": 32, "blondel": 45, "blue": 1, "bool": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44], "both": [0, 4, 24, 39, 49], "bound": [36, 37], "bradburi": 45, "bregman": [11, 27], "briefli": 49, "broadli": 0, "browser": 46, "bsd": [1, 3, 4, 5], "build": [39, 46, 49], "built": 49, "burdziak": 45, "c": [3, 4, 5, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 38, 39], "c_": [11, 12, 18, 26, 27, 30, 34, 35], "calcul": [9, 39, 42, 49], "call": [3, 14, 39, 49], "can": [0, 2, 3, 4, 5, 7, 12, 14, 15, 17, 20, 21, 23, 24, 28, 31, 32, 39, 40, 41, 44, 46, 49], "captur": [5, 12, 49], "care": [32, 47], "carr": 45, "case": [24, 28], "cassandra": 45, "cat": 1, "cell": [39, 45], "center": 49, "certain": 3, "cfg": 45, "chaffer": 45, "chanan": 45, "chang": [14, 46], "character": 49, "charlier": [45, 49], "check": [46, 48], "check_input": 14, "children": [10, 43], "choic": 32, "choos": [39, 49], "christin": 45, "citat": 48, "cite": 39, "class": [3, 4, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44, 49], "classic": 0, "claus": [1, 3, 4, 5], "clean": 48, "close": [4, 5, 49], "closer": 49, "closest": 12, "cluster": [1, 45], "cm": 1, "cmap": 1, "code": [1, 3, 4, 5, 7, 39, 46, 47, 48], "coeff_repuls": [15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "coeffici": [15, 17, 19, 23, 24, 28, 31, 32, 33, 35], "colab": 7, "collabor": 46, "collin": 45, "color": [1, 39], "column": 30, "com": [1, 3, 4, 5, 7, 39], "commit": 48, "common": [0, 39, 49], "commonli": 49, "commun": 46, "compar": [3, 47, 49], "comparison": 39, "compat": [39, 48], "complet": 46, "complex": [24, 28], "compliant": 46, "compon": [0, 1, 10, 14, 16, 22, 39, 43, 48, 49], "comput": [0, 3, 4, 9, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 45, 47, 49], "concentr": 12, "concern": 45, "conduct": [14, 48], "confer": 45, "connect": 1, "consequ": 1, "consid": [1, 5, 11, 12, 15, 17, 23, 27, 30, 31, 32, 34, 46], "consist": [11, 27, 48], "constant": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42, 49], "constraint": [12, 30], "construct": [0, 18, 39, 49], "contact": 39, "contain": 14, "context": 49, "contrast": [15, 33, 39, 45], "control": [1, 11, 19, 27, 30, 39, 48, 49], "convers": 5, "convex": [11, 12, 27, 30], "copi": 14, "core": [0, 46, 47], "correspond": [3, 12, 23, 49], "cost": [3, 12, 35, 49], "costli": 1, "courti": [39, 45], "cpu": [37, 41, 44, 49], "creat": [1, 39, 46], "criterion": [9, 19, 24, 28, 42], "cross": [3, 49], "cross_entropi": 3, "cross_entropy_loss": 3, "crucial": 49, "cuaz": 39, "cuda": [0, 14, 22, 25, 39, 41, 44, 47, 49], "current": [36, 37], "cusolv": [14, 22], "cut13": 45, "cuturi": [27, 45], "c\u00e9dric": 39, "d": [11, 27, 45], "d21": 33, "damrich": [15, 33, 45], "daniel": 45, "data": [0, 3, 4, 9, 10, 14, 16, 18, 22, 25, 26, 36, 37, 39, 41, 42, 43, 44, 45, 49], "dataset": [1, 4, 6, 7, 9, 10, 14, 19, 28, 31, 38, 39, 40, 42, 43, 47, 49], "david": 45, "dbohmhk22": 45, "deal": 49, "decid": 46, "decod": 3, "decompos": 14, "decomposit": [4, 39], "deep": 45, "def": [1, 3], "default": [9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 38, 41, 42, 43, 44], "defin": [0, 15, 17, 23, 31, 32, 33, 39, 49], "definit": 49, "degre": [21, 29, 35], "degrees_of_freedom": [21, 29, 35], "delimit": 3, "demo_ea_adapt": [1, 2, 40], "demo_ne_methods_affinity_match": [3, 6, 40], "demo_pca_via_affinity_match": [4, 6, 40], "demo_tsne_swiss_rol": [5, 6, 40], "denot": 49, "densiti": 1, "depend": [1, 5, 12, 49], "der": [12, 31, 45], "descent": [15, 17, 23, 31, 32, 33, 48], "describ": [30, 32, 46], "descript": 46, "design": [46, 47], "desir": [3, 46], "detach": 3, "detail": [46, 49], "determin": 49, "develop": [39, 46, 47], "deviat": 1, "devic": [0, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 37, 39, 41, 42, 43, 44, 47, 49], "dh21": 45, "diagon": [11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 41], "dict": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "dictat": 49, "dictionari": [11, 15, 17, 23, 27, 30, 31, 32, 33], "differ": [1, 15, 17, 23, 31, 32, 47, 49], "differenti": [9, 42], "diffus": 45, "digit": [4, 39], "dijk": [18, 45], "dimens": [9, 15, 17, 19, 20, 21, 23, 24, 26, 28, 31, 32, 33, 42, 45], "dimension": [9, 10, 12, 14, 42, 43, 45, 47], "direct": 12, "directli": [0, 39, 49], "directori": 46, "distanc": [11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 33, 34, 35, 38, 41, 44, 45], "distant": 5, "distribut": [12, 21, 29, 33, 35], "diverg": [27, 45], "divid": 14, "diwakar": 45, "dmitri": 45, "doc": 46, "docstr": 46, "document": 46, "doe": 14, "domain": [36, 37, 44, 49], "done": [0, 24, 27, 46], "doubl": 1, "doubli": [11, 27, 39, 45, 48, 49], "doublystochasticquadraticaffin": 49, "download": [1, 3, 4, 5, 7], "dr": [0, 10, 39, 43, 48, 49], "driven": 46, "drmodul": [0, 9, 14, 16, 22, 42, 49], "dtype": [1, 3, 36, 37], "dual": [11, 12, 27, 30, 32], "dubourg": 45, "duplic": 46, "durat": 19, "dure": [10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 41, 43, 44], "durif": 45, "e": [12, 14, 30, 45], "ea": 1, "each": [1, 10, 12, 14, 18, 24, 30, 39, 43, 49], "earli": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 46], "early_exagger": [3, 15, 17, 19, 23, 24, 28, 31, 32, 33], "early_exaggeration_it": [15, 17, 19, 23, 24, 28, 31, 32, 33], "easi": 39, "easili": [1, 39, 46], "ecosystem": 39, "edit": 46, "editor": 46, "edu": 5, "effect": [1, 12, 14, 15, 17, 23, 30, 31, 32, 34, 49], "effici": [39, 41, 44, 49], "effort": 46, "egg": 7, "eigendecomposit": [0, 48], "eigenvalu": [16, 39, 49], "eigenvector": [16, 49], "element": [11, 12, 14], "elev": 5, "elimin": 49, "ell": 12, "ell_2": [11, 49], "els": 5, "emb": 3, "embed": [4, 6, 7, 9, 10, 12, 15, 17, 19, 20, 23, 24, 28, 31, 32, 33, 35, 39, 40, 41, 42, 43, 44, 45, 47, 48], "embedding_": [3, 9, 42], "emphas": 47, "emploi": 1, "enabl": [3, 39, 49], "enable_metadata_rout": 14, "encod": [39, 49], "end": [1, 3, 4, 5, 7, 36, 37, 39, 49], "ensur": [1, 39, 47, 49], "entir": 49, "entrop": [2, 7, 12, 15, 17, 20, 23, 27, 30, 31, 32, 39, 40, 41, 44, 45, 48], "entropi": [1, 3, 12, 27, 30, 39, 48, 49], "entropicaffin": [1, 3, 15, 17, 23, 30, 31, 32], "enumer": [3, 5], "ep": [11, 27], "eps_squar": 30, "eps_square_affinity_in": 32, "eq": 27, "equival": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "error": [14, 48], "especi": 49, "essenti": 49, "estim": [14, 24, 49], "et": [11, 12, 15, 17, 18, 27, 30, 32, 33, 34, 35, 49], "euclidean": 33, "evalu": 49, "exact": 0, "exagger": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33], "exampl": [0, 1, 2, 3, 4, 5, 6, 7, 8, 39, 40], "except": 48, "execut": [2, 6, 8, 40], "exist": 14, "exp": [12, 13, 18, 20, 26, 27, 30, 34], "expon": 35, "express": [13, 21, 25, 29, 49], "extens": [0, 47], "extract": 14, "f": [3, 5, 11, 27, 36, 37], "fabian": 45, "factor": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 35, 42], "fail": 5, "fair": 47, "fals": [1, 3, 4, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 49], "famili": 49, "familiar": 46, "farther": 49, "fast": [24, 28], "faster": [14, 47], "featur": [14, 23, 49], "feedback": 46, "feel": 39, "fetch_openml": 39, "few": [1, 12], "feydi": [27, 45], "fig": [3, 5], "figsiz": [3, 5], "figur": [1, 3, 4, 5, 39, 49], "file": [2, 6, 8, 40, 46], "filter": 1, "final": 18, "find": [12, 36, 37], "first": [1, 4, 5, 9, 14, 18, 38, 42, 45, 46], "fit": [0, 3, 9, 10, 14, 16, 22, 35, 42, 43], "fit_transform": [4, 5, 9, 10, 14, 16, 22, 39, 42, 43, 49], "fix": [24, 27], "flamari": [5, 39, 45], "float": [3, 9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 42, 43], "float32": 39, "float64": [36, 37], "focu": [0, 47], "focus": 49, "follow": [0, 11, 13, 18, 27, 30, 35, 39, 46, 49], "fontsiz": 3, "foral": [11, 12, 27, 30, 34], "fork": 46, "form": [4, 9, 39, 42, 46, 49], "format": 39, "formul": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47], "formula": 49, "found": 49, "foundat": 49, "fr": 3, "frac": [12, 18, 21, 26, 29, 30], "framework": [0, 39, 47, 49], "fran": 45, "francisco": 45, "fred": [33, 45], "free": 39, "freedom": [21, 29, 35], "friendli": 39, "from": [0, 1, 2, 3, 4, 5, 6, 8, 14, 18, 26, 35, 39, 40, 45, 46, 49], "fsejournev": 45, "full": [1, 3, 4, 5, 46, 49], "fulli": 39, "function": [0, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 36, 37, 39, 42, 45, 46, 48, 49], "further": [33, 46], "futur": 46, "g": [1, 14], "ga": 45, "gabriel": 45, "galleri": [1, 3, 4, 5, 40], "gamma": [19, 24, 28, 49], "gap": 32, "garivi": 39, "gaussian": [3, 11, 13, 18, 20, 27, 39, 48], "gaussianaffin": [15, 16, 23, 49], "gen_batch": 14, "gene": 45, "gener": [0, 1, 3, 4, 5, 7, 14, 39, 46], "genom": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "geoffrei": 45, "geometri": [5, 39, 49], "geq": [12, 30], "gesvd": [14, 22], "gesvda": [14, 22], "gesvdj": [14, 22], "get": 3, "ghislain": 45, "gill": 45, "gimelshein": 45, "git": [7, 39], "github": [3, 7, 39, 46], "give": 0, "given": [1, 3, 4, 12, 14, 21, 25, 29, 30, 49], "glaun": 45, "global": [1, 4, 11, 27], "gmail": [1, 3, 4, 5], "go": [1, 3, 4, 5], "goal": 39, "good": 46, "googl": 7, "govern": [12, 49], "gpu": [0, 14, 39, 41, 44, 45, 47], "gradient": [0, 9, 27, 42, 48, 49], "gramfort": 45, "graph": [1, 49], "great": 46, "gregori": 45, "grisel": 45, "gross": 45, "ground": 12, "guid": 14, "h": [12, 27, 30, 39], "ha": [1, 14, 26, 49], "ham": [45, 49], "hamprecht": [33, 45], "have": [7, 39, 46], "heali": [33, 45], "henc": 49, "here": [1, 39, 49], "heterogen": 1, "heteroscedast": 1, "high": [1, 5, 45], "higher": 49, "highli": [39, 49], "hinton": [12, 23, 31, 45], "hlmscholkopf04": 45, "hm": 45, "homogen": 1, "hook": 48, "how": [4, 5, 14, 49], "howev": 0, "hr02": 45, "html": 46, "http": [3, 7, 39], "hugu": [1, 3, 4, 5, 39, 45], "hyperparamet": 49, "i": [0, 1, 3, 4, 5, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46, 47, 48, 49], "ichi": 45, "idea": 47, "ignor": [9, 10, 14, 42, 43], "ij": [11, 12, 15, 17, 18, 19, 23, 24, 26, 27, 28, 30, 31, 32, 33, 34, 35, 49], "illustr": 3, "imper": 45, "implement": [0, 1, 10, 11, 36, 37, 43, 49], "import": [1, 3, 4, 5, 39, 49], "imshow": 1, "includ": [39, 46, 48, 49], "incomplet": 46, "increas": [36, 37], "increment": [14, 48], "incrementalpca": 39, "incur": 49, "independ": 12, "indic": [39, 46, 49], "infer": 14, "influenc": 5, "inform": [10, 16, 22, 33, 41, 43, 44, 45, 46], "infotsn": [39, 48, 49], "inherit": [0, 49], "init": [4, 5, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "init_du": 11, "init_sc": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "initi": [9, 11, 15, 17, 19, 23, 24, 28, 31, 32, 33, 36, 37, 42, 49], "input": [0, 3, 4, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 25, 28, 31, 32, 33, 34, 36, 37, 39, 41, 42, 43, 44, 48, 49], "inria": 3, "insid": 14, "instal": [7, 46, 48], "instanc": [9, 42, 49], "instead": [14, 32], "int": [9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 42, 43], "intellig": 45, "interact": 45, "interestingli": 49, "intermedi": [11, 15, 17, 23, 27, 30, 31, 32, 33], "intern": 45, "interpol": 45, "interpret": 49, "introduc": [12, 15, 17, 18, 23, 26, 30, 31, 32, 33], "invers": 32, "involv": [15, 17, 23, 31, 32, 33, 49], "io": [3, 39], "ipca": 14, "ipynb": [1, 3, 4, 5], "issu": [1, 39, 46], "item": [1, 3, 46], "iter": [9, 11, 12, 14, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "its": [12, 18, 24, 39, 46, 48, 49], "j": [1, 11, 12, 15, 17, 23, 27, 30, 33, 34, 35, 45, 49], "jame": [33, 45], "jan": 45, "jean": 45, "ji": 18, "jian": 45, "jihun": 45, "jingzhou": 45, "joan": 45, "john": [33, 45], "journ": 45, "journal": 45, "juoza": 45, "jupyt": [1, 3, 4, 5, 7], "k": [1, 12, 18, 26, 27, 39, 48, 49], "kathail": 45, "keep": [12, 14], "keop": [0, 1, 4, 9, 10, 11, 12, 13, 15, 16, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 38, 39, 41, 42, 43, 44, 48], "kernel": [3, 11, 16, 18, 27, 39, 41, 44, 45, 48, 49], "kernelpca": 39, "kevin": 45, "keyword": [9, 14, 19, 22, 24, 28, 42], "killeen": 45, "kl": 27, "knopp": [27, 45], "knowledg": 46, "known": 49, "kobak": 45, "kristina": 45, "kullback": 27, "kwarg": [19, 31], "kwargs_affinity_out": [9, 19, 24, 28, 42], "kwargs_loss": [9, 42], "l": [9, 19, 24, 28, 42, 45, 49], "lambda": [19, 24, 28], "lambda_1": 49, "lambda_d": 49, "langl": [11, 12, 27, 30, 49], "laplacian": 39, "larg": [0, 12, 14, 41, 44, 45, 49], "larger": 12, "largest": 49, "largevi": [39, 48, 49], "last": 14, "latest": 39, "latter": [3, 49], "lauren": 45, "lazi": 38, "lazytensor": [0, 38], "lbfg": 30, "lead": 4, "learn": [9, 11, 14, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 39, 42, 45], "lee": 45, "left": [11, 18, 21, 26, 27, 29, 30, 35], "leftarrow": 18, "leibler": 27, "leland": [33, 45], "len": 5, "lerer": 45, "less": [1, 14], "let": [4, 5], "level": [2, 7, 12, 20, 40, 41, 44, 49], "leverag": [3, 14, 39, 47], "librari": [39, 45, 46], "licens": [1, 3, 4, 5], "lightspe": 45, "lihi": 45, "like": 48, "limit": [39, 49], "lin": 45, "linalg": [14, 22], "line": 7, "linear": [0, 15, 17, 23, 24, 28, 31, 32, 33], "link": [7, 46], "list": 46, "liu": 45, "lkopf": 45, "load": 7, "load_digit": 4, "load_numpy_from_url": 3, "local": [1, 46], "log": [9, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 42, 44, 49], "logaffin": [20, 21, 26, 27, 30, 49], "loss": [0, 3, 4, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42, 45, 49], "loss_fn": [3, 4, 9, 42], "low": [5, 39, 49], "lower": [36, 37], "lowrank": 14, "lowrank_nit": 14, "lowrank_q": 14, "lr": [3, 4, 9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42], "lr_affinity_in": 32, "luca": 45, "m": [1, 36, 37, 39, 46], "m18": 33, "m_sampl": 38, "maaten": [12, 31, 45], "machin": [45, 46], "magic": [18, 39, 48], "mai": [14, 30, 32], "main": 46, "maintain": 46, "make": [39, 46], "make_swiss_rol": 5, "manhattan": [15, 17, 23, 31, 32, 33], "mani": 49, "manifold": [0, 5, 33, 45], "manor": [26, 45], "manual_se": 1, "marco": 45, "margin": 49, "markov": 49, "mass": [12, 30, 49], "massa": 45, "massia": 39, "match": [3, 9, 39, 42, 49], "matcher": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "mathbb": [11, 12, 18, 26, 27, 30, 34, 36, 37], "mathbf": [9, 11, 12, 13, 15, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 42, 49], "mathcal": [9, 11, 19, 24, 27, 28, 42, 49], "mathemat": [45, 49], "mathieu": 45, "mathop": [11, 12, 27, 30], "mathrm": [11, 12, 19, 24, 27, 28, 30, 34, 49], "mathurin": 39, "matplotlib": [1, 3, 4, 5], "matric": [9, 11, 27, 41, 42, 44, 45], "matrix": [0, 1, 9, 10, 11, 12, 13, 16, 18, 19, 20, 21, 24, 25, 26, 27, 28, 29, 30, 34, 35, 38, 39, 41, 42, 43, 48, 49], "matsumoto": 45, "matur": 46, "max": 1, "max_it": [3, 4, 5, 9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "max_iter_affin": [15, 17, 23, 31, 33], "max_iter_affinity_in": 32, "maxim": 47, "maximum": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "mb": [2, 6, 8, 40], "mcinn": [33, 34, 35, 45, 49], "mean": 4, "mechan": 14, "meet": 46, "mei": 45, "melvil": [33, 45], "mem": [2, 6, 8, 40], "memori": [0, 14, 39, 45, 48], "mention": 46, "merg": 46, "meta": 14, "metadata": 14, "metadata_rout": 14, "method": [0, 4, 9, 10, 14, 19, 22, 24, 28, 36, 37, 39, 42, 43, 46, 47, 48], "method_nam": 3, "metric": [11, 12, 13, 15, 17, 18, 20, 21, 23, 26, 27, 29, 30, 31, 32, 33, 34, 35, 38, 41, 44], "metric_in": [15, 17, 23, 31, 32, 33], "metric_out": [15, 17, 23, 31, 32, 33], "mhm18": 45, "mi": 45, "michel": 45, "mika": 45, "min": [11, 12, 27, 30], "min_": [9, 19, 24, 28, 42, 49], "min_batch_s": 14, "min_dist": [33, 35], "min_j": 34, "ming": 45, "minibatch": 14, "minim": [0, 3, 12, 49], "minimum": [14, 33, 35], "minut": [1, 3, 4, 5], "misc": 39, "mistakenli": 5, "mmd": 45, "mnist": 39, "mnist_784": 39, "mode": 34, "model": [0, 4, 9, 10, 14, 16, 22, 42, 43], "modul": [16, 22, 39, 47], "modular": [0, 39, 47], "moon": 45, "mordant": 45, "more": [12, 30, 32, 49], "most": 49, "mrg": 46, "mu": 30, "much": 46, "multipl": [46, 48], "n": [3, 8, 11, 12, 14, 15, 17, 18, 26, 27, 30, 33, 36, 37, 39, 49], "n_": [12, 30, 34], "n_class": 4, "n_cluster": 1, "n_compon": [3, 4, 5, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 39, 42, 43], "n_featur": [9, 10, 14, 16, 22, 38, 42, 43], "n_neg": [15, 17, 24, 33], "n_neighbor": [33, 34, 49], "n_sampl": [5, 9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 38, 42, 43], "nadam": [11, 15, 17, 23, 30, 31, 32, 33], "naini": 45, "name": [14, 22], "natalia": 45, "ndarrai": [9, 10, 14, 16, 22, 42, 43], "ne": 49, "nearest": [12, 15, 17, 18, 23, 26, 30, 31, 32, 33, 34, 49], "necessari": 46, "need": 14, "neg": [0, 15, 17, 24, 33, 49], "neighbor": [1, 5, 6, 7, 9, 10, 12, 15, 17, 18, 19, 20, 23, 24, 26, 28, 30, 31, 32, 33, 34, 39, 40, 41, 42, 43, 44, 45, 47, 48], "neighborembed": [28, 32], "neighborhood": 49, "neirest": [18, 26], "neural": [33, 45], "neurip": 33, "new": [0, 14, 39, 47, 49], "nicola": [39, 45], "nikla": 45, "nn": [12, 39, 48], "no_grad": 27, "nodiag": 16, "nois": [2, 5, 7, 12, 15, 20, 33, 40, 41, 44, 49], "none": [1, 3, 9, 10, 11, 14, 15, 17, 18, 19, 22, 23, 24, 26, 28, 31, 32, 33, 35, 36, 37, 38, 42, 43], "nonneg": 45, "norm": [11, 49], "normal": [1, 3, 4, 11, 15, 17, 18, 20, 21, 23, 26, 27, 31, 32, 33, 39, 48, 49], "normalization_dim": [1, 3, 20, 21, 26], "normalizedgaussianaffin": [1, 3, 49], "notabl": 49, "note": [0, 14, 19], "notebook": [1, 3, 4, 5, 7], "notimplementederror": [10, 43], "notion": 0, "now": [1, 4], "np": [3, 9, 10, 14, 16, 22, 42, 43], "nu": [21, 29], "nuanc": 49, "number": [1, 9, 10, 11, 12, 14, 15, 16, 17, 19, 22, 23, 24, 27, 28, 30, 31, 32, 33, 34, 39, 42, 43, 46, 49], "numpi": [0, 3, 39], "numpy_arrai": 3, "object": [0, 1, 4, 9, 14, 16, 19, 24, 28, 32, 39, 42, 48, 49], "observ": [1, 5], "obtain": 4, "occur": 1, "offer": [39, 49], "often": 49, "oi": 45, "olivi": 45, "onc": 46, "one": [12, 32, 49], "ones": [11, 12, 27, 30, 49], "onli": [1, 5, 12, 14, 22, 49], "onto": [10, 11, 14, 16, 22, 27, 43], "open": [39, 46], "oper": [0, 41, 44, 45], "optim": [0, 3, 9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 39, 42, 45, 49], "optimizer_kwarg": [3, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "optimum": 12, "option": [9, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 44], "order": [9, 42], "origin": [14, 46], "ot": 32, "other": [14, 45, 46], "otherwis": 14, "out": [1, 48], "outlin": [46, 49], "output": [0, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 39, 42, 46, 48, 49], "output_dim": [15, 17, 23, 31, 32, 33], "outsid": 49, "overflow": [0, 39, 45], "overridden": [10, 43], "overwritten": 14, "p": [11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33], "p_": [12, 15, 17, 18, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34], "pacif": 45, "pairwis": [0, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 38, 39, 41, 44, 49], "paper": 49, "paralleliz": 49, "param": 3, "paramet": [0, 1, 5, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 49], "parent": 49, "part": 49, "partial": 14, "partial_fit": 14, "particular": 0, "particularli": 49, "pass": [14, 46], "paszk": [45, 49], "pattabiraman": 45, "paul": 45, "pca": [5, 6, 7, 9, 10, 14, 15, 17, 19, 23, 24, 25, 28, 31, 32, 33, 39, 40, 41, 42, 43, 47, 48, 49], "pedregosa": [45, 49], "per": 1, "perform": [0, 4, 9, 10, 11, 12, 14, 16, 22, 27, 36, 37, 42, 43, 45, 49], "perona": [26, 45], "perplex": [1, 12, 15, 17, 23, 30, 31, 32, 34, 39, 49], "perplexity_valu": 5, "peter": 45, "peyr": 45, "pgm": 45, "phase": [15, 17, 19, 23, 24, 28, 31, 32, 33], "pick": 46, "pietro": 45, "pip": [7, 39, 47], "pipelin": [14, 47, 49], "place": 46, "plai": 49, "plan": 46, "pleas": [14, 39, 46], "plot": 1, "plot_affinity_graph": 1, "plt": [1, 3, 4, 5], "pmlr": 45, "point": [0, 1, 5, 12, 15, 17, 18, 24, 26, 27, 33, 35, 38, 46, 49], "polytechniqu": 5, "pooja": 45, "popular": 49, "posit": [36, 37, 49], "possibl": [0, 46], "power": 47, "pr": [46, 48], "practic": [30, 32], "pre": 48, "precis": [11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 34, 36, 37], "precomput": [9, 10, 42, 43], "prefer": 46, "prefix": 46, "prepar": 46, "preprint": [33, 45], "present": [0, 49], "preserv": [39, 49], "prettenhof": 45, "prevent": [46, 49], "previous": 14, "princip": [14, 16, 22, 49], "print": [3, 10, 16, 22, 36, 37, 41, 43, 44, 49], "prior": 46, "problem": [0, 9, 11, 12, 19, 24, 27, 28, 30, 39, 42, 49], "problemat": 49, "proceed": 45, "process": [9, 14, 19, 24, 28, 33, 42, 45], "product": [4, 25, 39, 48, 49], "progress": 46, "proj": [11, 27], "project": [5, 10, 11, 14, 16, 22, 27, 33, 39, 43, 45, 46, 48], "properti": 1, "propos": 49, "provid": [0, 9, 14, 35, 39, 42, 47, 49], "public": 46, "pull": [46, 48], "purpos": 46, "pvg": 45, "py": [1, 2, 3, 4, 5, 6, 40], "pykeop": [38, 39, 49], "pypi": 39, "pyplot": [1, 3, 4, 5], "pytest": 46, "python": [1, 3, 4, 5, 7, 39, 45, 46], "pytorch": [0, 9, 14, 15, 17, 23, 30, 31, 32, 33, 39, 42, 45], "q": [15, 17, 19, 23, 24, 27, 28, 31, 32, 33], "q_": [15, 17, 19, 23, 24, 27, 28, 31, 32, 33], "qiaozhu": 45, "quad": [11, 12, 27, 30, 34, 49], "quadrat": [11, 39, 48, 49], "qualiti": 48, "question": 39, "r": [11, 12, 18, 26, 27, 30, 34, 36, 37, 39, 45], "rais": [10, 14, 43, 48], "randn": 49, "random": [9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43, 49], "random_st": [5, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43], "rang": [1, 5, 39, 49], "rangl": [11, 12, 27, 30, 49], "rate": [9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42], "read": 3, "readi": 46, "receiv": 49, "recent": 49, "recommend": [3, 12], "recov": [45, 49], "reduc": [3, 24], "reduct": [9, 10, 14, 33, 42, 43, 45, 47], "refer": [33, 39, 49], "regardless": 1, "region": 1, "regular": [11, 27], "regularis": 45, "relat": [12, 30], "relev": 14, "reli": [0, 39, 49], "remark": 1, "remedi": 1, "remi": 5, "remov": 16, "rep": [19, 24, 28, 49], "replac": 49, "repositori": 46, "repres": 49, "represent": [39, 49], "reproduc": [3, 9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43, 47], "repuls": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "request": [3, 14, 46, 48], "requir": [32, 39, 46, 49], "research": [39, 45], "resolv": 46, "resort": 32, "respect": [1, 49], "respons": 3, "rest": 47, "restrict": [36, 37], "result": [11, 15, 17, 23, 27, 30, 31, 32, 33, 46, 49], "retain": 14, "return": [3, 9, 14, 16, 22, 36, 37, 38, 42], "review": 46, "rho_i": 34, "richard": 45, "right": [11, 18, 21, 26, 27, 29, 30, 35], "role": 49, "roll": [6, 7, 9, 10, 19, 28, 31, 40, 42, 43, 47], "ron": 45, "root": [0, 12, 15, 17, 23, 31, 34, 36, 37, 46], "roshan": 45, "rotat": 4, "rout": 14, "row": [3, 12, 18, 30, 49], "rowei": [12, 23, 45], "rsinghlab": 3, "run": [0, 1, 4, 5, 7, 46, 47], "r\u00e9mi": [5, 39], "sam": 45, "same": [0, 1, 3, 4], "sampl": [0, 11, 12, 14, 15, 17, 18, 23, 24, 26, 27, 30, 31, 32, 33, 49], "sampledneighborembed": [0, 15, 17, 33], "satur": 12, "scalar": [4, 25, 39, 48, 49], "scalarproductaffin": [0, 4, 49], "scale": [0, 9, 12, 15, 17, 19, 23, 24, 28, 31, 32, 33, 41, 42, 44, 45, 49], "scatter": [1, 3, 4, 5], "sch": 45, "schedul": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "scheduler_kwarg": [9, 17, 19, 23, 24, 28, 31, 32, 33, 42], "schiebing": 45, "scikit": [14, 39, 45], "score": [39, 48], "scot": 3, "script": [1, 3, 4, 5], "se": 30, "sea": 30, "seamlessli": [0, 39, 47], "search": [0, 12, 15, 17, 23, 31, 34, 36, 37], "sebastian": [33, 45], "second": [1, 3, 4, 5, 38], "section": 0, "see": [4, 14, 49], "seed": [9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43], "seek": 46, "select": [12, 15, 17, 23, 30, 31, 32, 33], "self": [9, 14, 16, 22, 26, 39, 42, 45, 48], "semi": 49, "sequenc": 14, "serv": [46, 49], "set": [0, 1, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 26, 27, 29, 30, 32, 33, 34, 35, 41, 47, 49], "set_config": 14, "set_fit_request": 14, "set_partial_fit_request": 14, "set_titl": [3, 5], "set_xtick": 3, "set_ytick": 3, "sgd": [11, 15, 17, 23, 30, 31, 32, 33], "shannon": [12, 27, 30, 49], "shape": [9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 36, 37, 38, 42, 43, 49], "share": [47, 49], "sharma": 45, "short": 5, "should": [10, 14, 24, 28, 36, 37, 43, 46, 49], "show": [1, 4, 5], "shun": 45, "sigma": [1, 13, 18, 20, 26, 34], "sigma_i": [18, 26], "sigma_j": 26, "signific": 46, "significantli": [3, 5, 15, 17, 23, 31, 32], "silhouett": [39, 48], "similar": [0, 34, 39, 49], "simpli": 12, "simplifi": 39, "simul": 1, "sinc": 12, "singl": [39, 45], "singular": 4, "sinkhorn": [27, 45], "sinkhornaffin": [32, 49], "size": [1, 14, 36, 37, 49], "sk67": 45, "sklearn": [4, 5, 14, 39, 49], "slice": 14, "small": [12, 49], "smallest": 12, "snare_rna": 3, "snareseq_typ": 3, "sne": [3, 31, 39, 45, 48, 49], "sne_affinity_match": 3, "snekhorn": [12, 39, 45, 48, 49], "soft": [12, 49], "sole": [39, 49], "solut": [4, 30, 49], "solv": [0, 9, 11, 12, 19, 24, 27, 28, 32, 39, 42, 48, 49], "solver": 0, "some": 14, "someth": 46, "soon": 46, "sourc": [1, 3, 4, 5, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46], "space": [0, 5, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 35, 42, 49], "spars": [1, 28, 45, 49], "sparselogaffin": [12, 24, 28, 34], "sparseneighborembed": [0, 23, 24, 31], "sparsiti": [0, 1, 3, 12, 24, 34, 39], "speak": 0, "specif": [0, 3], "specifi": [20, 21], "spectral": [4, 39, 45, 48], "speed": 39, "sphinx": [1, 3, 4, 5, 7], "sphinxcontrib": 48, "split": 3, "spread": [12, 33, 35, 49], "sqeuclidean": [11, 12, 13, 15, 17, 18, 20, 21, 23, 26, 27, 29, 30, 31, 32, 33, 34, 35, 38, 41, 44], "sqrt": 49, "squar": [4, 30, 32, 49], "square_loss": [4, 9, 42], "stabl": [30, 32], "standard": [1, 46], "star": [11, 12, 27, 30, 49], "star_": 30, "star_i": [11, 27, 30, 34], "star_j": [11, 27, 30], "start": 46, "static": 14, "statist": 45, "statu": 46, "std": 1, "stephen": 45, "stochast": [11, 23, 24, 27, 31, 39, 45, 48, 49], "stop": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 36, 37, 42], "store": [11, 15, 17, 23, 27, 30, 31, 32, 33, 49], "str": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 37, 38, 41, 42, 43, 44], "straightforward": 49, "strategi": [0, 39], "student": [11, 21, 27, 29, 33, 35, 39, 48, 49], "studentaffin": [17, 31, 49], "studi": 33, "style": [45, 46, 48], "sub": 14, "subclass": [10, 43], "submit": 46, "subplot": [1, 5], "subproblem": 12, "subspac": 14, "suggest": 39, "sum": 49, "sum_": [12, 15, 17, 18, 19, 23, 24, 27, 28, 30, 31, 32, 33, 49], "sum_i": [15, 23, 49], "sum_j": [34, 49], "summar": 49, "support": [0, 39], "sure": 46, "svd": [6, 7, 9, 10, 14, 22, 25, 40, 41, 42, 43, 47], "svd_driver": [14, 22], "svd_lowrank": 14, "swiss": [6, 7, 9, 10, 19, 28, 31, 40, 42, 43, 47], "switch": 48, "symbol": 39, "symmetr": [11, 12, 27, 30, 32, 39, 45, 48, 49], "symmetri": [30, 49], "symmetric_affin": 32, "symmetricentropicaffin": [12, 32, 49], "system": [33, 45], "t": [3, 5, 12, 18, 21, 29, 30, 31, 33, 35, 39, 45, 48], "take": [3, 49], "taken": [46, 47, 49], "tang": [17, 45], "target": 4, "task": 46, "techniqu": 39, "templat": 48, "tensor": [0, 1, 9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 36, 37, 38, 39, 42, 43], "term": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "termin": 47, "test": 46, "tetsuya": 45, "text": [5, 11, 12, 27, 30, 34, 46, 49], "text2d": 5, "th": [18, 26], "than": [14, 49], "thank": [39, 46], "thei": [0, 49], "them": [3, 7], "therebi": 49, "therefor": 3, "thi": [0, 1, 10, 11, 14, 22, 24, 27, 28, 32, 39, 43, 46, 49], "thibault": 45, "thirion": 45, "those": [0, 49], "three": 1, "threshold": [11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 34, 36, 37], "through": 39, "thu": [1, 12, 49], "tight_layout": 3, "time": [0, 1, 3, 4, 5, 11, 12, 27, 30], "titl": [1, 4, 5, 39, 46], "titouan": [3, 39, 45], "tlzm16": 45, "toi": 1, "tol": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "tol_affin": [15, 17, 23, 31, 33], "tol_affinity_in": 32, "toler": [9, 19, 24, 28, 42], "tolog": [9, 11, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 42], "too": [5, 12, 14, 32], "toolbox": 39, "top": [7, 11, 12, 25, 27, 30, 49], "torch": [1, 9, 10, 11, 14, 15, 16, 17, 22, 23, 27, 31, 32, 33, 36, 37, 38, 42, 43], "torchdr": [1, 4, 5, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46, 49], "total": [1, 2, 3, 4, 5, 6, 8, 40], "track": [27, 46], "tracker": [39, 46], "train": 14, "transax": 5, "transform": [5, 9, 10, 14, 16, 22, 39, 42, 43], "transformermixin": 49, "transit": 49, "transport": [11, 12, 27, 32, 45], "transpos": [12, 18], "trevor": 45, "triplet": 49, "trouv": 45, "true": [0, 9, 10, 11, 12, 13, 14, 18, 19, 20, 21, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 45, 49], "try": 46, "tsne": [0, 1, 3, 6, 7, 9, 10, 12, 19, 28, 30, 39, 40, 42, 43, 47, 49], "tsnekhorn": [12, 49], "tune": [26, 39, 45, 48], "tupl": [20, 21, 26], "twenti": 45, "two": [0, 3, 9, 38, 42, 49], "two_sne_dict": 3, "txt": 3, "type": [9, 14, 16, 22, 36, 37, 39, 42, 46], "typic": [41, 44, 49], "u": 4, "ultim": 49, "umap": [3, 34, 35, 39, 45, 48, 49], "umapaffinityin": [33, 49], "umapaffinityout": [33, 49], "unchang": 14, "under": 49, "uniform": [33, 45], "uniformli": 49, "unit": 12, "unlik": 30, "unnorm": 16, "unnormalizedaffin": [24, 25, 28, 35], "unnormalizedlogaffin": [13, 29], "unrol": 32, "unstabl": 32, "up": 4, "updat": [14, 30, 32, 46], "upper": [36, 37], "url": [3, 39], "url_i": 3, "url_x": 3, "urllib": 3, "urlopen": 3, "us": [0, 1, 4, 11, 13, 14, 15, 16, 17, 18, 21, 26, 27, 29, 30, 32, 34, 35, 38, 39, 45, 46], "usag": 3, "user": [14, 39], "usual": [39, 49], "utf": 3, "util": [14, 49], "v": 49, "valid": 14, "valu": [4, 12, 15, 17, 23, 30, 31, 32], "van": [1, 3, 4, 5, 12, 18, 30, 31, 32, 39, 45, 49], "vanassel2024torchdr": 39, "vanasselhugu": [1, 3, 4, 5], "varepsilon": [11, 12, 27, 30], "varepsilon_i": 12, "vari": [2, 7, 12, 20, 40, 41, 44, 49], "variabl": [11, 12, 27, 30, 32, 34], "varieti": 0, "variou": [0, 1, 39, 47], "varoquaux": 45, "vavfc24": 45, "vayer": [3, 39, 45], "vdmh08": 45, "vdsn": 45, "vector": [11, 12, 27, 30], "verbos": [1, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 41, 42, 43, 44], "veri": 1, "version": [12, 14, 30, 39], "via": [0, 3, 6, 7, 9, 10, 22, 25, 27, 34, 39, 40, 41, 42, 43, 47, 48, 49], "vialard": 45, "view": [45, 49], "view_init": 5, "viewabl": 46, "vincent": [39, 45], "visual": 45, "wai": [0, 11, 27, 39, 46, 47], "we": [1, 3, 4, 5, 46, 49], "web": [45, 46], "weight": 49, "weiss": 45, "welcom": 46, "well": [0, 4, 39, 49], "what": 49, "when": [3, 5, 14, 24, 28, 32, 46, 48, 49], "where": [9, 11, 12, 13, 15, 17, 19, 20, 21, 24, 25, 27, 28, 29, 30, 33, 34, 35, 39, 42, 46, 49], "whether": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44, 49], "which": [10, 11, 12, 14, 15, 16, 17, 19, 20, 21, 22, 23, 26, 27, 30, 31, 32, 33, 36, 37, 39, 43, 49], "while": [0, 12, 47, 49], "wide": [39, 45, 49], "wip": 46, "wise": [12, 18, 26, 30], "with_grad": 27, "within": 49, "without": [0, 39, 45, 46, 49], "work": [0, 14, 22, 46, 47], "workflow": 46, "world": 45, "write": [11, 27], "written": 39, "x": [1, 3, 4, 5, 9, 10, 14, 16, 18, 22, 25, 26, 38, 39, 42, 43, 49], "x1": 1, "x2": 1, "x3": 1, "x_": 39, "x_embed": 5, "x_new": [14, 16, 22], "xavier": 45, "xi": [12, 30], "y": [3, 9, 10, 38, 42, 43], "year": 39, "yield": 14, "yim": 45, "you": [0, 7, 14, 39, 46, 47], "your": [0, 39, 46, 47], "z": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 39, 42, 49], "z_am": 4, "z_gpu": 39, "z_svd": 4, "zelnik": [26, 45], "zeme": 45, "zero": [13, 16, 18, 20, 21, 26, 29, 34, 35, 41], "zero_diag": [1, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 41, 44], "zhang": [11, 45, 49], "zip": [1, 3, 4, 5, 7], "zmms23": 45, "zmp04": 45, "\u00e7": 45, "\u00e9": 45, "\u00eb": 45, "\u00f6": 45}, "titles": ["API and Modules", "Entropic Affinities can adapt to varying noise levels", "Computation times", "Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation", "PCA via SVD and via AffinityMatcher", "TSNE embedding of the swiss roll dataset", "Computation times", "Gallery", "Computation times", "AffinityMatcher", "DRModule", "DoublyStochasticQuadraticAffinity", "EntropicAffinity", "GaussianAffinity", "IncrementalPCA", "InfoTSNE", "KernelPCA", "LargeVis", "MAGICAffinity", "NeighborEmbedding", "NormalizedGaussianAffinity", "NormalizedStudentAffinity", "PCA", "SNE", "SampledNeighborEmbedding", "ScalarProductAffinity", "SelfTuningAffinity", "SinkhornAffinity", "SparseNeighborEmbedding", "StudentAffinity", "SymmetricEntropicAffinity", "TSNE", "TSNEkhorn", "UMAP", "UMAPAffinityIn", "UMAPAffinityOut", "binary_search", "false_position", "pairwise_distances", "Torch Dimensionality Reduction", "Computation times", "Affinity", "AffinityMatcher", "DRModule", "LogAffinity", "Bibliography", "How to Contribute", "Quick Start Guide", "Releases", "User Guide"], "titleterms": {"0": 48, "09": 48, "1": 48, "17": 48, "2024": 48, "On": 3, "adapt": 1, "advanc": 0, "affin": [0, 1, 3, 7, 39, 41, 49], "affinitymatch": [3, 4, 9, 42, 47], "algorithm": 39, "api": [0, 3], "automat": 49, "avoid": 49, "bandwidth": 1, "base": [0, 49], "basic": 7, "bibliographi": 45, "binary_search": 36, "can": 1, "cell": 3, "citat": 39, "class": 0, "compat": 0, "comput": [2, 5, 6, 8, 40], "constant": 1, "content": 49, "contribut": 46, "contributor": 46, "dataset": [3, 5], "date": 39, "differ": 3, "differenti": 49, "dimension": [0, 39, 49], "directli": 3, "distanc": 0, "doublystochasticquadraticaffin": 11, "drmodul": [10, 43], "effici": 3, "embed": [0, 3, 5, 49], "entrop": [0, 1, 49], "entropicaffin": [12, 49], "equival": 3, "estim": 0, "evalu": 39, "exampl": [9, 10, 12, 19, 20, 22, 23, 24, 25, 28, 31, 33, 41, 42, 43, 44, 47, 49], "express": 3, "false_posit": 37, "featur": 39, "find": 39, "formul": [3, 49], "galleri": 7, "gaussian": 1, "gaussianaffin": 13, "gene": 3, "gener": 49, "genom": 3, "get": 39, "gpu": 49, "guid": [47, 49], "help": 39, "how": 46, "imag": [4, 5], "impact": 5, "implement": 39, "incrementalpca": 14, "infotsn": 15, "instal": [39, 47], "keop": 49, "kernelpca": 16, "knn": 0, "label": 3, "largevi": 17, "lazi": 49, "level": 1, "load": [3, 4, 5], "logaffin": 44, "magicaffin": 18, "matcher": 3, "memori": 49, "method": [3, 49], "metric": 39, "modul": [0, 49], "neighbor": [0, 3, 49], "neighborembed": 19, "new": 46, "nois": 1, "normal": 0, "normalis": 1, "normalizedgaussianaffin": 20, "normalizedstudentaffin": 21, "other": 49, "overflow": 49, "overview": 49, "pairwise_dist": 38, "pca": [4, 22], "perplex": 5, "plot": [3, 5], "project": 49, "quadrat": 0, "quick": 47, "rather": 3, "reduct": [0, 39, 49], "releas": 48, "roll": 5, "row": 1, "run": 3, "sampledneighborembed": 24, "scalarproductaffin": 25, "see": 5, "selftuningaffin": 26, "seq": 3, "simpl": 0, "sinkhornaffin": 27, "sklearn": 0, "snare": 3, "sne": 23, "sparseneighborembed": 28, "spectral": [0, 49], "spotlight": 49, "start": [39, 47], "structur": 49, "studentaffin": 29, "support": 49, "svd": 4, "swiss": 5, "symbol": 49, "symmetricentropicaffin": 30, "tabl": 49, "tensor": 49, "than": 3, "time": [2, 6, 8, 40], "tip": 46, "toi": [4, 5], "torch": [39, 49], "torchdr": [0, 3, 47], "tsne": [5, 31], "tsnekhorn": 32, "type": 3, "umap": [0, 33], "umapaffinityin": 34, "umapaffinityout": 35, "us": [3, 9, 10, 12, 19, 20, 22, 23, 24, 25, 28, 31, 33, 41, 42, 43, 44, 47, 49], "user": 49, "util": 0, "vari": 1, "variou": 49, "version": 48, "via": 4, "when": 47}})
\ No newline at end of file
+Search.setIndex({"alltitles": {"API and Modules": [[0, null]], "Advanced Dimensionality Reduction with TorchDR": [[0, "advanced-dimensionality-reduction-with-torchdr"]], "Affinities": [[7, "affinities"], [39, "affinities"], [49, "affinities"]], "Affinities Normalized by kNN Distances": [[0, "affinities-normalized-by-knn-distances"]], "Affinity": [[41, null]], "Affinity Classes": [[0, "affinity-classes"]], "AffinityMatcher": [[9, null], [42, null]], "Avoiding memory overflows with KeOps symbolic (lazy) tensors": [[49, "avoiding-memory-overflows-with-keops-symbolic-lazy-tensors"]], "Base Classes": [[0, "base-classes"]], "Base Neighbor Embedding Modules": [[0, "base-neighbor-embedding-modules"]], "Base structure": [[49, "base-structure"]], "Basics": [[7, "basics"]], "Bibliography": [[45, null]], "Citation": [[39, "citation"]], "Computation times": [[2, null], [6, null], [8, null], [40, null]], "Compute the TSNE embedding": [[5, "compute-the-tsne-embedding"]], "DRModule": [[10, null], [43, null]], "Dimensionality Reduction Algorithms": [[39, "dimensionality-reduction-algorithms"]], "Dimensionality Reduction Modules": [[49, "dimensionality-reduction-modules"]], "Dimensionality Reduction sklearn Compatible Estimators": [[0, "dimensionality-reduction-sklearn-compatible-estimators"]], "DoublyStochasticQuadraticAffinity": [[11, null]], "Entropic Affinities": [[0, "entropic-affinities"]], "Entropic Affinities can adapt to varying noise levels": [[1, null]], "Entropic affinity (adaptive bandwidth)": [[1, "entropic-affinity-adaptive-bandwidth"]], "EntropicAffinity": [[12, null]], "Evaluation Metric": [[39, "evaluation-metric"]], "Examples using Affinity:": [[41, "examples-using-affinity"]], "Examples using AffinityMatcher:": [[9, "examples-using-affinitymatcher"], [42, "examples-using-affinitymatcher"], [47, "examples-using-affinitymatcher"]], "Examples using DRModule:": [[10, "examples-using-drmodule"], [43, "examples-using-drmodule"]], "Examples using EntropicAffinity:": [[12, "examples-using-entropicaffinity"], [49, "examples-using-entropicaffinity"]], "Examples using LogAffinity:": [[44, "examples-using-logaffinity"]], "Examples using NeighborEmbedding:": [[19, "examples-using-neighborembedding"]], "Examples using NormalizedGaussianAffinity:": [[20, "examples-using-normalizedgaussianaffinity"]], "Examples using PCA:": [[22, "examples-using-pca"]], "Examples using SNE:": [[23, "examples-using-sne"]], "Examples using SampledNeighborEmbedding:": [[24, "examples-using-sampledneighborembedding"]], "Examples using ScalarProductAffinity:": [[25, "examples-using-scalarproductaffinity"]], "Examples using SparseNeighborEmbedding:": [[28, "examples-using-sparseneighborembedding"]], "Examples using TSNE:": [[31, "examples-using-tsne"]], "Examples using UMAP:": [[33, "examples-using-umap"]], "Finding Help": [[39, "finding-help"]], "Gallery": [[7, null]], "GaussianAffinity": [[13, null]], "General Formulation of Dimensionality Reduction": [[49, "general-formulation-of-dimensionality-reduction"]], "Getting Started": [[39, "getting-started"]], "How to Contribute": [[46, null]], "Implemented Features (to date)": [[39, "implemented-features-to-date"]], "IncrementalPCA": [[14, null]], "InfoTSNE": [[15, null]], "Installation": [[39, "installation"], [47, "installation"]], "KernelPCA": [[16, null]], "LargeVis": [[17, null]], "Load the SNARE-seq dataset (gene expression) with cell type labels": [[3, "load-the-snare-seq-dataset-gene-expression-with-cell-type-labels"]], "Load toy images": [[4, "load-toy-images"], [5, "load-toy-images"]], "LogAffinity": [[44, null]], "MAGICAffinity": [[18, null]], "Neighbor Embedding": [[0, "neighbor-embedding"], [49, "neighbor-embedding"]], "Neighbor Embedding on genomics & equivalent affinity matcher formulation": [[3, null]], "NeighborEmbedding": [[19, null]], "New contributor tips": [[46, "new-contributor-tips"]], "NormalizedGaussianAffinity": [[20, null]], "NormalizedStudentAffinity": [[21, null]], "On the efficiency of using the torchdr API rather than AffinityMatcher directly": [[3, "on-the-efficiency-of-using-the-torchdr-api-rather-than-affinitymatcher-directly"]], "Other various affinities": [[49, "other-various-affinities"]], "Overview": [[49, "overview"]], "PCA": [[22, null]], "PCA via AffinityMatcher": [[4, "pca-via-affinitymatcher"]], "PCA via SVD": [[4, "pca-via-svd"]], "PCA via SVD and via AffinityMatcher": [[4, null]], "Plot the TSNE embedding": [[5, "plot-the-tsne-embedding"]], "Plot the dataset": [[5, "plot-the-dataset"]], "Plot the different embeddings": [[3, "plot-the-different-embeddings"]], "Quadratic Affinities": [[0, "quadratic-affinities"]], "Quick Start Guide": [[47, null]], "Releases": [[48, null]], "Row-normalised Gaussian affinity with constant bandwidth": [[1, "row-normalised-gaussian-affinity-with-constant-bandwidth"]], "Run neighbor embedding methods": [[3, "run-neighbor-embedding-methods"]], "SNE": [[23, null]], "SampledNeighborEmbedding": [[24, null]], "ScalarProductAffinity": [[25, null]], "See the impact of perplexity": [[5, "see-the-impact-of-perplexity"]], "SelfTuningAffinity": [[26, null]], "Simple Affinities": [[0, "simple-affinities"]], "SinkhornAffinity": [[27, null]], "SparseNeighborEmbedding": [[28, null]], "Spectral Embedding": [[0, "spectral-embedding"]], "Spectral methods": [[49, "spectral-methods"]], "Spotlight on affinities based on entropic projections": [[49, "spotlight-on-affinities-based-on-entropic-projections"]], "StudentAffinity": [[29, null]], "SymmetricEntropicAffinity": [[30, null]], "TSNE": [[31, null]], "TSNE embedding of the swiss roll dataset": [[5, null]], "TSNEkhorn": [[32, null]], "Table of Contents": [[49, "table-of-contents"]], "Torch Dimensionality Reduction": [[39, null]], "Torch GPU support and automatic differentiation": [[49, "torch-gpu-support-and-automatic-differentiation"]], "UMAP": [[33, null]], "UMAP Affinities": [[0, "umap-affinities"]], "UMAPAffinityIn": [[34, null]], "UMAPAffinityOut": [[35, null]], "User Guide": [[49, null]], "Using AffinityMatcher": [[3, "using-affinitymatcher"]], "Utils": [[0, "utils"]], "Version 0.1 (2024-09-17)": [[48, "version-0-1-2024-09-17"]], "When to use TorchDR": [[47, "when-to-use-torchdr"]], "binary_search": [[36, null]], "false_position": [[37, null]], "pairwise_distances": [[38, null]]}, "docnames": ["all", "auto_examples/affinities/demo_ea_adaptivity", "auto_examples/affinities/sg_execution_times", "auto_examples/basics/demo_ne_methods_affinity_matcher", "auto_examples/basics/demo_pca_via_affinity_matcher", "auto_examples/basics/demo_tsne_swiss_roll", "auto_examples/basics/sg_execution_times", "auto_examples/index", "auto_examples/sg_execution_times", "gen_modules/torchdr.AffinityMatcher", "gen_modules/torchdr.DRModule", "gen_modules/torchdr.DoublyStochasticQuadraticAffinity", "gen_modules/torchdr.EntropicAffinity", "gen_modules/torchdr.GaussianAffinity", "gen_modules/torchdr.IncrementalPCA", "gen_modules/torchdr.InfoTSNE", "gen_modules/torchdr.KernelPCA", "gen_modules/torchdr.LargeVis", "gen_modules/torchdr.MAGICAffinity", "gen_modules/torchdr.NeighborEmbedding", "gen_modules/torchdr.NormalizedGaussianAffinity", "gen_modules/torchdr.NormalizedStudentAffinity", "gen_modules/torchdr.PCA", "gen_modules/torchdr.SNE", "gen_modules/torchdr.SampledNeighborEmbedding", "gen_modules/torchdr.ScalarProductAffinity", "gen_modules/torchdr.SelfTuningAffinity", "gen_modules/torchdr.SinkhornAffinity", "gen_modules/torchdr.SparseNeighborEmbedding", "gen_modules/torchdr.StudentAffinity", "gen_modules/torchdr.SymmetricEntropicAffinity", "gen_modules/torchdr.TSNE", "gen_modules/torchdr.TSNEkhorn", "gen_modules/torchdr.UMAP", "gen_modules/torchdr.UMAPAffinityIn", "gen_modules/torchdr.UMAPAffinityOut", "gen_modules/torchdr.binary_search", "gen_modules/torchdr.false_position", "gen_modules/torchdr.pairwise_distances", "index", "sg_execution_times", "stubs/torchdr.Affinity", "stubs/torchdr.AffinityMatcher", "stubs/torchdr.DRModule", "stubs/torchdr.LogAffinity", "torchdr.bibliography", "torchdr.contributing", "torchdr.quick_start", "torchdr.releases", "torchdr.user_guide"], "envversion": {"sphinx": 64, "sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx.ext.viewcode": 1, "sphinxcontrib.bibtex": 9}, "filenames": ["all.rst", "auto_examples/affinities/demo_ea_adaptivity.rst", "auto_examples/affinities/sg_execution_times.rst", "auto_examples/basics/demo_ne_methods_affinity_matcher.rst", "auto_examples/basics/demo_pca_via_affinity_matcher.rst", "auto_examples/basics/demo_tsne_swiss_roll.rst", "auto_examples/basics/sg_execution_times.rst", "auto_examples/index.rst", "auto_examples/sg_execution_times.rst", "gen_modules/torchdr.AffinityMatcher.rst", "gen_modules/torchdr.DRModule.rst", "gen_modules/torchdr.DoublyStochasticQuadraticAffinity.rst", "gen_modules/torchdr.EntropicAffinity.rst", "gen_modules/torchdr.GaussianAffinity.rst", "gen_modules/torchdr.IncrementalPCA.rst", "gen_modules/torchdr.InfoTSNE.rst", "gen_modules/torchdr.KernelPCA.rst", "gen_modules/torchdr.LargeVis.rst", "gen_modules/torchdr.MAGICAffinity.rst", "gen_modules/torchdr.NeighborEmbedding.rst", "gen_modules/torchdr.NormalizedGaussianAffinity.rst", "gen_modules/torchdr.NormalizedStudentAffinity.rst", "gen_modules/torchdr.PCA.rst", "gen_modules/torchdr.SNE.rst", "gen_modules/torchdr.SampledNeighborEmbedding.rst", "gen_modules/torchdr.ScalarProductAffinity.rst", "gen_modules/torchdr.SelfTuningAffinity.rst", "gen_modules/torchdr.SinkhornAffinity.rst", "gen_modules/torchdr.SparseNeighborEmbedding.rst", "gen_modules/torchdr.StudentAffinity.rst", "gen_modules/torchdr.SymmetricEntropicAffinity.rst", "gen_modules/torchdr.TSNE.rst", "gen_modules/torchdr.TSNEkhorn.rst", "gen_modules/torchdr.UMAP.rst", "gen_modules/torchdr.UMAPAffinityIn.rst", "gen_modules/torchdr.UMAPAffinityOut.rst", "gen_modules/torchdr.binary_search.rst", "gen_modules/torchdr.false_position.rst", "gen_modules/torchdr.pairwise_distances.rst", "index.rst", "sg_execution_times.rst", "stubs/torchdr.Affinity.rst", "stubs/torchdr.AffinityMatcher.rst", "stubs/torchdr.DRModule.rst", "stubs/torchdr.LogAffinity.rst", "torchdr.bibliography.rst", "torchdr.contributing.rst", "torchdr.quick_start.rst", "torchdr.releases.rst", "torchdr.user_guide.rst"], "indexentries": {"affinity (class in torchdr)": [[41, "torchdr.Affinity", false]], "affinitymatcher (class in torchdr)": [[9, "torchdr.AffinityMatcher", false], [42, "torchdr.AffinityMatcher", false]], "binary_search() (in module torchdr)": [[36, "torchdr.binary_search", false]], "doublystochasticquadraticaffinity (class in torchdr)": [[11, "torchdr.DoublyStochasticQuadraticAffinity", false]], "drmodule (class in torchdr)": [[10, "torchdr.DRModule", false], [43, "torchdr.DRModule", false]], "entropicaffinity (class in torchdr)": [[12, "torchdr.EntropicAffinity", false]], "false_position() (in module torchdr)": [[37, "torchdr.false_position", false]], "fit() (torchdr.affinitymatcher method)": [[9, "torchdr.AffinityMatcher.fit", false], [42, "torchdr.AffinityMatcher.fit", false]], "fit() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.fit", false]], "fit() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.fit", false]], "fit() (torchdr.pca method)": [[22, "torchdr.PCA.fit", false]], "fit_transform() (torchdr.affinitymatcher method)": [[9, "torchdr.AffinityMatcher.fit_transform", false], [42, "torchdr.AffinityMatcher.fit_transform", false]], "fit_transform() (torchdr.drmodule method)": [[10, "torchdr.DRModule.fit_transform", false], [43, "torchdr.DRModule.fit_transform", false]], "fit_transform() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.fit_transform", false]], "fit_transform() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.fit_transform", false]], "fit_transform() (torchdr.pca method)": [[22, "torchdr.PCA.fit_transform", false]], "gaussianaffinity (class in torchdr)": [[13, "torchdr.GaussianAffinity", false]], "gen_batches() (torchdr.incrementalpca static method)": [[14, "torchdr.IncrementalPCA.gen_batches", false]], "incrementalpca (class in torchdr)": [[14, "torchdr.IncrementalPCA", false]], "infotsne (class in torchdr)": [[15, "torchdr.InfoTSNE", false]], "kernelpca (class in torchdr)": [[16, "torchdr.KernelPCA", false]], "largevis (class in torchdr)": [[17, "torchdr.LargeVis", false]], "logaffinity (class in torchdr)": [[44, "torchdr.LogAffinity", false]], "magicaffinity (class in torchdr)": [[18, "torchdr.MAGICAffinity", false]], "module": [[0, "module-torchdr", false], [47, "module-torchdr", false], [49, "module-torchdr", false]], "neighborembedding (class in torchdr)": [[19, "torchdr.NeighborEmbedding", false]], "normalizedgaussianaffinity (class in torchdr)": [[20, "torchdr.NormalizedGaussianAffinity", false]], "normalizedstudentaffinity (class in torchdr)": [[21, "torchdr.NormalizedStudentAffinity", false]], "pairwise_distances() (in module torchdr)": [[38, "torchdr.pairwise_distances", false]], "partial_fit() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.partial_fit", false]], "pca (class in torchdr)": [[22, "torchdr.PCA", false]], "sampledneighborembedding (class in torchdr)": [[24, "torchdr.SampledNeighborEmbedding", false]], "scalarproductaffinity (class in torchdr)": [[25, "torchdr.ScalarProductAffinity", false]], "selftuningaffinity (class in torchdr)": [[26, "torchdr.SelfTuningAffinity", false]], "set_fit_request() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.set_fit_request", false]], "set_partial_fit_request() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.set_partial_fit_request", false]], "sinkhornaffinity (class in torchdr)": [[27, "torchdr.SinkhornAffinity", false]], "sne (class in torchdr)": [[23, "torchdr.SNE", false]], "sparseneighborembedding (class in torchdr)": [[28, "torchdr.SparseNeighborEmbedding", false]], "studentaffinity (class in torchdr)": [[29, "torchdr.StudentAffinity", false]], "symmetricentropicaffinity (class in torchdr)": [[30, "torchdr.SymmetricEntropicAffinity", false]], "torchdr": [[0, "module-torchdr", false], [47, "module-torchdr", false], [49, "module-torchdr", false]], "transform() (torchdr.incrementalpca method)": [[14, "torchdr.IncrementalPCA.transform", false]], "transform() (torchdr.kernelpca method)": [[16, "torchdr.KernelPCA.transform", false]], "transform() (torchdr.pca method)": [[22, "torchdr.PCA.transform", false]], "tsne (class in torchdr)": [[31, "torchdr.TSNE", false]], "tsnekhorn (class in torchdr)": [[32, "torchdr.TSNEkhorn", false]], "umap (class in torchdr)": [[33, "torchdr.UMAP", false]], "umapaffinityin (class in torchdr)": [[34, "torchdr.UMAPAffinityIn", false]], "umapaffinityout (class in torchdr)": [[35, "torchdr.UMAPAffinityOut", false]]}, "objects": {"": [[49, 0, 0, "-", "torchdr"]], "torchdr": [[41, 1, 1, "", "Affinity"], [42, 1, 1, "", "AffinityMatcher"], [43, 1, 1, "", "DRModule"], [11, 1, 1, "", "DoublyStochasticQuadraticAffinity"], [12, 1, 1, "", "EntropicAffinity"], [13, 1, 1, "", "GaussianAffinity"], [14, 1, 1, "", "IncrementalPCA"], [15, 1, 1, "", "InfoTSNE"], [16, 1, 1, "", "KernelPCA"], [17, 1, 1, "", "LargeVis"], [44, 1, 1, "", "LogAffinity"], [18, 1, 1, "", "MAGICAffinity"], [19, 1, 1, "", "NeighborEmbedding"], [20, 1, 1, "", "NormalizedGaussianAffinity"], [21, 1, 1, "", "NormalizedStudentAffinity"], [22, 1, 1, "", "PCA"], [23, 1, 1, "", "SNE"], [24, 1, 1, "", "SampledNeighborEmbedding"], [25, 1, 1, "", "ScalarProductAffinity"], [26, 1, 1, "", "SelfTuningAffinity"], [27, 1, 1, "", "SinkhornAffinity"], [28, 1, 1, "", "SparseNeighborEmbedding"], [29, 1, 1, "", "StudentAffinity"], [30, 1, 1, "", "SymmetricEntropicAffinity"], [31, 1, 1, "", "TSNE"], [32, 1, 1, "", "TSNEkhorn"], [33, 1, 1, "", "UMAP"], [34, 1, 1, "", "UMAPAffinityIn"], [35, 1, 1, "", "UMAPAffinityOut"], [36, 3, 1, "", "binary_search"], [37, 3, 1, "", "false_position"], [38, 3, 1, "", "pairwise_distances"]], "torchdr.AffinityMatcher": [[42, 2, 1, "", "fit"], [42, 2, 1, "", "fit_transform"]], "torchdr.DRModule": [[43, 2, 1, "", "fit_transform"]], "torchdr.IncrementalPCA": [[14, 2, 1, "", "fit"], [14, 2, 1, "", "fit_transform"], [14, 2, 1, "", "gen_batches"], [14, 2, 1, "", "partial_fit"], [14, 2, 1, "", "set_fit_request"], [14, 2, 1, "", "set_partial_fit_request"], [14, 2, 1, "", "transform"]], "torchdr.KernelPCA": [[16, 2, 1, "", "fit"], [16, 2, 1, "", "fit_transform"], [16, 2, 1, "", "transform"]], "torchdr.PCA": [[22, 2, 1, "", "fit"], [22, 2, 1, "", "fit_transform"], [22, 2, 1, "", "transform"]]}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "function", "Python function"]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:function"}, "terms": {"": [3, 4, 5, 7, 9, 12, 14, 30, 33, 42, 45], "0": [1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 40, 42, 43], "00": [2, 6, 8, 40], "000": 8, "0001": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "001": [12, 15, 17, 23, 30, 31, 32, 33], "01": [6, 40], "01816": 45, "02": [6, 40], "03": [2, 40], "03426": [33, 45], "05": [5, 6, 11, 27, 34, 40], "054": 40, "07": [9, 15, 17, 19, 23, 24, 28, 31, 33, 42], "09": [36, 37], "09816": 45, "1": [1, 2, 3, 4, 5, 9, 11, 12, 13, 14, 15, 17, 19, 20, 21, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 35, 42, 45, 46, 49], "10": [1, 3, 5, 23, 32, 33], "100": [3, 15, 17, 23, 31, 32, 33, 49], "1000": [9, 11, 12, 27, 34, 36, 37, 42], "11": 45, "111": 5, "12": [3, 5, 15, 17, 31, 45], "125": 48, "127": 48, "130": 48, "137": 48, "14": [3, 5, 6, 40], "143": 48, "147": 48, "15": 3, "166": [1, 2, 40], "174": 45, "18": 45, "1802": [33, 45], "19": 45, "1967": [27, 45], "1d": 36, "1e": [9, 11, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "1e0": [3, 9, 19, 24, 28, 32, 42], "1e1": 4, "2": [1, 3, 4, 5, 9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 21, 22, 23, 24, 28, 29, 30, 31, 32, 33, 42, 43, 45, 46, 49], "20": [1, 5], "200": 5, "2000": [15, 19, 23, 24, 28, 31, 32, 33], "2002": [12, 23, 45], "2004": [26, 45, 49], "2008": [12, 31, 45], "2011": [45, 49], "2013": [27, 45], "2016": [17, 45], "2018": [18, 33, 34, 35, 45, 49], "2019": [27, 45, 49], "2021": [33, 45, 49], "2022": [15, 45], "2023": [11, 45, 49], "2024": [12, 30, 32, 39, 45, 49], "21": 45, "22": 45, "2206": 45, "22nd": 45, "2307": 45, "24": 3, "25": 27, "250": [15, 17, 23, 31, 32, 33], "25th": 45, "2681": 45, "2690": 45, "2825": 45, "2830": 45, "287": 45, "297": 45, "3": [1, 3, 4, 5, 6, 12, 14, 45, 46], "30": [12, 15, 17, 23, 30, 31, 32, 33, 34, 39], "3000": 17, "34": [6, 33, 45], "343": 45, "348": 45, "351": [4, 6, 40], "38": 40, "3d": 5, "4": [3, 5, 9, 14, 15, 17, 19, 24, 28, 31, 32, 33, 40, 42], "47": 45, "5": [1, 4, 5, 14, 15, 17, 24, 30], "50": [4, 5, 15, 39], "500": [5, 30], "5798": 45, "5809": 45, "6": [1, 5, 45], "66": 5, "668": [5, 6, 40], "7": [9, 15, 17, 18, 19, 24, 26, 28, 31, 33, 42], "716": 45, "729": 45, "74": 45, "8": [3, 5], "869": [3, 6, 40], "888": 6, "A": [8, 12, 14, 24, 39, 45, 46, 48, 49], "As": 1, "By": [1, 15, 17, 23, 31, 32, 33], "For": [0, 14, 47, 49], "If": [9, 12, 14, 19, 24, 27, 28, 32, 38, 39, 41, 42, 44, 46, 49], "In": [0, 1, 3, 4, 12, 18, 24, 26, 28, 45, 49], "It": [0, 9, 12, 14, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 39, 42, 49], "Its": [13, 21, 25, 29, 35, 39], "On": [33, 45], "Such": 1, "The": [0, 1, 3, 9, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 30, 31, 32, 33, 34, 36, 37, 38, 39, 41, 42, 44, 45, 46, 49], "Then": 18, "These": [0, 49], "To": [1, 3, 39, 47, 49], "With": 12, "_": [5, 11, 12, 19, 24, 27, 28, 30, 36, 37, 49], "_1": 49, "_2": 11, "_d": 49, "_i": [18, 26, 49], "_j": 49, "_n": 49, "_type_": [15, 17, 31, 32], "a_x": [9, 42, 49], "a_z": [9, 42, 49], "abc": [10, 41, 43], "abil": 0, "abov": [18, 26, 32, 49], "abstract": [10, 43], "acceler": [0, 14, 39, 47, 49], "accept": 39, "accord": [20, 21], "account": 46, "acknowledg": 46, "across": [1, 12, 47], "act": 12, "activ": 46, "ad": [7, 14, 30, 48], "adam": [3, 9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42, 45], "adapt": [2, 7, 12, 14, 20, 39, 40, 41, 44, 48, 49], "add": 0, "add_ax": 5, "add_subplot": [3, 5], "addit": [9, 19, 24, 28, 41, 42, 44], "addition": 49, "address": 46, "adequ": 14, "advanc": [33, 45], "aff": 1, "aff_ea": 1, "affin": [2, 4, 6, 9, 10, 11, 12, 13, 15, 16, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 40, 42, 43, 44, 45, 47, 48], "affinity_in": [3, 4, 9, 19, 24, 28, 42, 49], "affinity_matrix": 49, "affinity_out": [3, 4, 9, 19, 24, 28, 42, 49], "affinitymatch": [0, 6, 7, 10, 19, 22, 25, 40, 41, 43, 48], "after": [14, 19, 46], "aim": [39, 49], "akin": 12, "al": [11, 12, 15, 17, 18, 27, 30, 32, 33, 34, 35, 49], "alain": 45, "alexandr": 45, "alexi": 45, "algorithm": [1, 3, 11, 12, 15, 17, 20, 23, 27, 30, 31, 32, 33, 34, 36, 37, 49], "alia": 14, "all": [0, 7, 11, 12, 27, 30, 39, 40, 46, 47, 48, 49], "all_method": 3, "allow": [0, 14, 30, 35, 46, 47, 49], "along": [18, 20, 21, 26], "alpha": [1, 5], "also": [12, 46], "amari": 45, "ambient": 5, "ambros": 45, "among": 12, "amount": [9, 19, 24, 28, 30, 42], "an": [1, 3, 14, 24, 28, 36, 37, 39, 45, 46, 48, 49], "analog": 49, "analysi": [14, 16, 22, 49], "ani": [39, 46, 48, 49], "anoth": [11, 27, 49], "antiga": 45, "apart": 35, "api": [39, 46, 49], "append": 5, "appli": 14, "applic": 48, "approach": [4, 47, 49], "appropri": 46, "approv": 46, "approxim": [0, 14, 33, 45, 49], "ar": [0, 5, 10, 12, 14, 16, 22, 27, 35, 39, 43, 46, 47, 49], "arg": [11, 12, 27, 30], "argument": [9, 14, 15, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42], "around": [1, 49], "arrai": [3, 39], "artifici": 45, "arxiv": [33, 45], "ascent": [11, 27, 30], "aspect": 49, "assel": [1, 3, 4, 5, 12, 30, 32, 39, 45, 49], "assist": 46, "associ": [3, 39, 49], "astyp": 39, "attract": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "aur\u00e9lien": 39, "author": [1, 3, 4, 5, 39], "authorship": 46, "auto": [9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44], "auto_exampl": [2, 6, 8], "auto_examples_jupyt": 7, "auto_examples_python": 7, "autodiff": 45, "automat": [9, 42, 46], "avail": [1, 3, 4, 12, 14, 22, 39, 49], "averag": [12, 18], "avoid": 39, "awai": 49, "ax": [3, 5], "azim": 5, "b": [33, 35, 45], "back": 46, "backend": [39, 48], "balanc": 49, "bandwidth": [13, 18, 20, 26, 49], "base": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 39, 41, 42, 43, 44, 48], "base_kernel": [11, 27, 49], "baseestim": [10, 43, 49], "basic": [3, 6, 40, 48, 49], "batch": [14, 36, 37, 39], "batch_siz": 14, "becaus": 1, "been": 49, "befor": 46, "begin": [36, 37], "benchmark": 47, "benefit": [39, 49], "benjamin": 45, "bernhard": 45, "bertrand": 45, "best": [39, 49], "between": [0, 3, 11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 38, 45, 49], "bibtex": [39, 48], "big": [12, 15, 23, 31], "binari": [36, 49], "black": [1, 46], "block": [39, 49], "blog": 32, "blondel": 45, "blue": 1, "bool": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44], "both": [0, 4, 24, 39, 49], "bound": [36, 37], "bradburi": 45, "bregman": [11, 27], "briefli": 49, "broadli": 0, "browser": 46, "bsd": [1, 3, 4, 5], "build": [39, 46, 49], "built": 49, "burdziak": 45, "c": [3, 4, 5, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 38, 39], "c_": [11, 12, 18, 26, 27, 30, 34, 35], "calcul": [9, 39, 42, 49], "call": [3, 14, 39, 49], "can": [0, 2, 3, 4, 5, 7, 12, 14, 15, 17, 20, 21, 23, 24, 28, 31, 32, 39, 40, 41, 44, 46, 49], "captur": [5, 12, 49], "care": [32, 47], "carr": 45, "case": [24, 28], "cassandra": 45, "cat": 1, "cell": [39, 45], "center": 49, "certain": 3, "cfg": 45, "chaffer": 45, "chanan": 45, "chang": [14, 46], "character": 49, "charlier": [45, 49], "check": [46, 48], "check_input": 14, "children": [10, 43], "choic": 32, "choos": [39, 49], "christin": 45, "citat": 48, "cite": 39, "class": [3, 4, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44, 49], "classic": 0, "claus": [1, 3, 4, 5], "clean": 48, "close": [4, 5, 49], "closer": 49, "closest": 12, "cluster": [1, 45], "cm": 1, "cmap": 1, "code": [1, 3, 4, 5, 7, 39, 46, 47, 48], "coeff_repuls": [15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "coeffici": [15, 17, 19, 23, 24, 28, 31, 32, 33, 35], "colab": 7, "collabor": 46, "collin": 45, "color": [1, 39], "column": 30, "com": [1, 3, 4, 5, 7, 39], "commit": 48, "common": [0, 39, 49], "commonli": 49, "commun": 46, "compar": [3, 47, 49], "comparison": 39, "compat": [39, 48], "complet": 46, "complex": [24, 28], "compliant": 46, "compon": [0, 1, 10, 14, 16, 22, 39, 43, 48, 49], "comput": [0, 3, 4, 9, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 45, 47, 49], "concentr": 12, "concern": 45, "conduct": [14, 48], "confer": 45, "connect": 1, "consequ": 1, "consid": [1, 5, 11, 12, 15, 17, 23, 27, 30, 31, 32, 34, 46], "consist": [11, 27, 48], "constant": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42, 49], "constraint": [12, 30], "construct": [0, 18, 39, 49], "contact": 39, "contain": 14, "context": 49, "contrast": [15, 33, 39, 45], "control": [1, 11, 19, 27, 30, 39, 48, 49], "convers": 5, "convex": [11, 12, 27, 30], "copi": 14, "core": [0, 46, 47], "correspond": [3, 12, 23, 49], "cost": [3, 12, 35, 49], "costli": 1, "courti": [39, 45], "cpu": [37, 41, 44, 49], "creat": [1, 39, 46], "criterion": [9, 19, 24, 28, 42], "cross": [3, 49], "cross_entropi": 3, "cross_entropy_loss": 3, "crucial": 49, "cuaz": 39, "cuda": [0, 14, 22, 25, 39, 41, 44, 47, 49], "current": [36, 37], "cusolv": [14, 22], "cut13": 45, "cuturi": [27, 45], "c\u00e9dric": 39, "d": [11, 27, 45], "d21": 33, "damrich": [15, 33, 45], "daniel": 45, "data": [0, 3, 4, 9, 10, 14, 16, 18, 22, 25, 26, 36, 37, 39, 41, 42, 43, 44, 45, 49], "dataset": [1, 4, 6, 7, 9, 10, 14, 19, 28, 31, 38, 39, 40, 42, 43, 47, 49], "david": 45, "dbohmhk22": 45, "deal": 49, "decid": 46, "decod": 3, "decompos": 14, "decomposit": [4, 39], "deep": 45, "def": [1, 3], "default": [9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 38, 41, 42, 43, 44], "defin": [0, 15, 17, 23, 31, 32, 33, 39, 49], "definit": 49, "degre": [21, 29, 35], "degrees_of_freedom": [21, 29, 35], "delimit": 3, "demo_ea_adapt": [1, 2, 40], "demo_ne_methods_affinity_match": [3, 6, 40], "demo_pca_via_affinity_match": [4, 6, 40], "demo_tsne_swiss_rol": [5, 6, 40], "denot": 49, "densiti": 1, "depend": [1, 5, 12, 49], "der": [12, 31, 45], "descent": [15, 17, 23, 31, 32, 33, 48], "describ": [30, 32, 46], "descript": 46, "design": [46, 47], "desir": [3, 46], "detach": 3, "detail": [46, 49], "determin": 49, "develop": [39, 46, 47], "deviat": 1, "devic": [0, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 37, 39, 41, 42, 43, 44, 47, 49], "dh21": 45, "diagon": [11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 41], "dict": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "dictat": 49, "dictionari": [11, 15, 17, 23, 27, 30, 31, 32, 33], "differ": [1, 15, 17, 23, 31, 32, 47, 49], "differenti": [9, 42], "diffus": 45, "digit": [4, 39], "dijk": [18, 45], "dimens": [9, 15, 17, 19, 20, 21, 23, 24, 26, 28, 31, 32, 33, 42, 45], "dimension": [9, 10, 12, 14, 42, 43, 45, 47], "direct": 12, "directli": [0, 39, 49], "directori": 46, "distanc": [11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 33, 34, 35, 38, 41, 44, 45], "distant": 5, "distribut": [12, 21, 29, 33, 35], "diverg": [27, 45], "divid": 14, "diwakar": 45, "dmitri": 45, "doc": 46, "docstr": 46, "document": 46, "doe": 14, "domain": [36, 37, 44, 49], "done": [0, 24, 27, 46], "doubl": 1, "doubli": [11, 27, 39, 45, 48, 49], "doublystochasticquadraticaffin": 49, "download": [1, 3, 4, 5, 7], "dr": [0, 10, 39, 43, 48, 49], "driven": 46, "drmodul": [0, 9, 14, 16, 22, 42, 49], "dtype": [1, 3, 36, 37], "dual": [11, 12, 27, 30, 32], "dubourg": 45, "duplic": 46, "durat": 19, "dure": [10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 41, 43, 44], "durif": 45, "e": [12, 14, 30, 45], "ea": 1, "each": [1, 10, 12, 14, 18, 24, 30, 39, 43, 49], "earli": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 46], "early_exagger": [3, 15, 17, 19, 23, 24, 28, 31, 32, 33], "early_exaggeration_it": [15, 17, 19, 23, 24, 28, 31, 32, 33], "easi": 39, "easili": [1, 39, 46], "ecosystem": 39, "edit": 46, "editor": 46, "edu": 5, "effect": [1, 12, 14, 15, 17, 23, 30, 31, 32, 34, 49], "effici": [39, 41, 44, 49], "effort": 46, "egg": 7, "eigendecomposit": [0, 48], "eigenvalu": [16, 39, 49], "eigenvector": [16, 49], "element": [11, 12, 14], "elev": 5, "elimin": 49, "ell": 12, "ell_2": [11, 49], "els": 5, "emb": 3, "embed": [4, 6, 7, 9, 10, 12, 15, 17, 19, 20, 23, 24, 28, 31, 32, 33, 35, 39, 40, 41, 42, 43, 44, 45, 47, 48], "embedding_": [3, 9, 42], "emphas": 47, "emploi": 1, "enabl": [3, 39, 49], "enable_metadata_rout": 14, "encod": [39, 49], "end": [1, 3, 4, 5, 7, 36, 37, 39, 49], "ensur": [1, 39, 47, 49], "entir": 49, "entrop": [2, 7, 12, 15, 17, 20, 23, 27, 30, 31, 32, 39, 40, 41, 44, 45, 48], "entropi": [1, 3, 12, 27, 30, 39, 48, 49], "entropicaffin": [1, 3, 15, 17, 23, 30, 31, 32], "enumer": [3, 5], "ep": [11, 27], "eps_squar": 30, "eps_square_affinity_in": 32, "eq": 27, "equival": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "error": [14, 48], "especi": 49, "essenti": 49, "estim": [14, 24, 49], "et": [11, 12, 15, 17, 18, 27, 30, 32, 33, 34, 35, 49], "euclidean": 33, "evalu": 49, "exact": 0, "exagger": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33], "exampl": [0, 1, 2, 3, 4, 5, 6, 7, 8, 39, 40], "except": 48, "execut": [2, 6, 8, 40], "exist": 14, "exp": [12, 13, 18, 20, 26, 27, 30, 34], "expon": 35, "express": [13, 21, 25, 29, 49], "extens": [0, 47], "extract": 14, "f": [3, 5, 11, 27, 36, 37], "fabian": 45, "factor": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 35, 42], "fail": 5, "fair": 47, "fals": [1, 3, 4, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 49], "famili": 49, "familiar": 46, "farther": 49, "fast": [24, 28], "faster": [14, 47], "featur": [14, 23, 49], "feedback": 46, "feel": 39, "fetch_openml": 39, "few": [1, 12], "feydi": [27, 45], "fig": [3, 5], "figsiz": [3, 5], "figur": [1, 3, 4, 5, 39, 49], "file": [2, 6, 8, 40, 46], "filter": 1, "final": 18, "find": [12, 36, 37], "first": [1, 4, 5, 9, 14, 18, 38, 42, 45, 46], "fit": [0, 3, 9, 10, 14, 16, 22, 35, 42, 43], "fit_transform": [4, 5, 9, 10, 14, 16, 22, 39, 42, 43, 49], "fix": [24, 27], "flamari": [5, 39, 45], "float": [3, 9, 10, 11, 12, 13, 14, 15, 16, 17, 19, 20, 21, 22, 23, 24, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 42, 43], "float32": 39, "float64": [36, 37], "focu": [0, 47], "focus": 49, "follow": [0, 11, 13, 18, 27, 30, 35, 39, 46, 49], "fontsiz": 3, "foral": [11, 12, 27, 30, 34], "fork": 46, "form": [4, 9, 39, 42, 46, 49], "format": 39, "formul": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47], "formula": 49, "found": 49, "foundat": 49, "fr": 3, "frac": [12, 18, 21, 26, 29, 30], "framework": [0, 39, 47, 49], "fran": 45, "francisco": 45, "fred": [33, 45], "free": 39, "freedom": [21, 29, 35], "friendli": 39, "from": [0, 1, 2, 3, 4, 5, 6, 8, 14, 18, 26, 35, 39, 40, 45, 46, 49], "fsejournev": 45, "full": [1, 3, 4, 5, 46, 49], "fulli": 39, "function": [0, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 36, 37, 39, 42, 45, 46, 48, 49], "further": [33, 46], "futur": 46, "g": [1, 14], "ga": 45, "gabriel": 45, "galleri": [1, 3, 4, 5, 40], "gamma": [19, 24, 28, 49], "gap": 32, "garivi": 39, "gaussian": [3, 11, 13, 18, 20, 27, 39, 48], "gaussianaffin": [15, 16, 23, 49], "gen_batch": 14, "gene": 45, "gener": [0, 1, 3, 4, 5, 7, 14, 39, 46], "genom": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "geoffrei": 45, "geometri": [5, 39, 49], "geq": [12, 30], "gesvd": [14, 22], "gesvda": [14, 22], "gesvdj": [14, 22], "get": 3, "ghislain": 45, "gill": 45, "gimelshein": 45, "git": [7, 39], "github": [3, 7, 39, 46], "give": 0, "given": [1, 3, 4, 12, 14, 21, 25, 29, 30, 49], "glaun": 45, "global": [1, 4, 11, 27], "gmail": [1, 3, 4, 5], "go": [1, 3, 4, 5], "goal": 39, "good": 46, "googl": 7, "govern": [12, 49], "gpu": [0, 14, 39, 41, 44, 45, 47], "gradient": [0, 9, 27, 42, 48, 49], "gramfort": 45, "graph": [1, 49], "great": 46, "gregori": 45, "grisel": 45, "gross": 45, "ground": 12, "guid": 14, "h": [12, 27, 30, 39], "ha": [1, 14, 26, 49], "ham": [45, 49], "hamprecht": [33, 45], "have": [7, 39, 46], "heali": [33, 45], "henc": 49, "here": [1, 39, 49], "heterogen": 1, "heteroscedast": 1, "high": [1, 5, 45], "higher": 49, "highli": [39, 49], "hinton": [12, 23, 31, 45], "hlmscholkopf04": 45, "hm": 45, "homogen": 1, "hook": 48, "how": [4, 5, 14, 49], "howev": 0, "hr02": 45, "html": 46, "http": [3, 7, 39], "hugu": [1, 3, 4, 5, 39, 45], "hyperparamet": 49, "i": [0, 1, 3, 4, 5, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46, 47, 48, 49], "ichi": 45, "idea": 47, "ignor": [9, 10, 14, 42, 43], "ij": [11, 12, 15, 17, 18, 19, 23, 24, 26, 27, 28, 30, 31, 32, 33, 34, 35, 49], "illustr": 3, "imper": 45, "implement": [0, 1, 10, 11, 36, 37, 43, 49], "import": [1, 3, 4, 5, 39, 49], "imshow": 1, "includ": [39, 46, 48, 49], "incomplet": 46, "increas": [36, 37], "increment": [14, 48], "incrementalpca": 39, "incur": 49, "independ": 12, "indic": [39, 46, 49], "infer": 14, "influenc": 5, "inform": [10, 16, 22, 33, 41, 43, 44, 45, 46], "infotsn": [39, 48, 49], "inherit": [0, 49], "init": [4, 5, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "init_du": 11, "init_sc": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "initi": [9, 11, 15, 17, 19, 23, 24, 28, 31, 32, 33, 36, 37, 42, 49], "input": [0, 3, 4, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 25, 28, 31, 32, 33, 34, 36, 37, 39, 41, 42, 43, 44, 48, 49], "inria": 3, "insid": 14, "instal": [7, 46, 48], "instanc": [9, 42, 49], "instead": [14, 32], "int": [9, 10, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 42, 43], "intellig": 45, "interact": 45, "interestingli": 49, "intermedi": [11, 15, 17, 23, 27, 30, 31, 32, 33], "intern": 45, "interpol": 45, "interpret": 49, "introduc": [12, 15, 17, 18, 23, 26, 30, 31, 32, 33], "invers": 32, "involv": [15, 17, 23, 31, 32, 33, 49], "io": [3, 39], "ipca": 14, "ipynb": [1, 3, 4, 5], "issu": [1, 39, 46], "item": [1, 3, 46], "iter": [9, 11, 12, 14, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "its": [12, 18, 24, 39, 46, 48, 49], "j": [1, 11, 12, 15, 17, 23, 27, 30, 33, 34, 35, 45, 49], "jame": [33, 45], "jan": 45, "jean": 45, "ji": 18, "jian": 45, "jihun": 45, "jingzhou": 45, "joan": 45, "john": [33, 45], "journ": 45, "journal": 45, "juoza": 45, "jupyt": [1, 3, 4, 5, 7], "k": [1, 12, 18, 26, 27, 39, 48, 49], "kathail": 45, "keep": [12, 14], "keop": [0, 1, 4, 9, 10, 11, 12, 13, 15, 16, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 38, 39, 41, 42, 43, 44, 48], "kernel": [3, 11, 16, 18, 27, 39, 41, 44, 45, 48, 49], "kernelpca": 39, "kevin": 45, "keyword": [9, 14, 19, 22, 24, 28, 42], "killeen": 45, "kl": 27, "knopp": [27, 45], "knowledg": 46, "known": 49, "kobak": 45, "kristina": 45, "kullback": 27, "kwarg": [19, 31], "kwargs_affinity_out": [9, 19, 24, 28, 42], "kwargs_loss": [9, 42], "l": [9, 19, 24, 28, 42, 45, 49], "lambda": [19, 24, 28], "lambda_1": 49, "lambda_d": 49, "langl": [11, 12, 27, 30, 49], "laplacian": 39, "larg": [0, 12, 14, 41, 44, 45, 49], "larger": 12, "largest": 49, "largevi": [39, 48, 49], "last": 14, "latest": 39, "latter": [3, 49], "lauren": 45, "lazi": 38, "lazytensor": [0, 38], "lbfg": 30, "lead": 4, "learn": [9, 11, 14, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 39, 42, 45], "lee": 45, "left": [11, 18, 21, 26, 27, 29, 30, 35], "leftarrow": 18, "leibler": 27, "leland": [33, 45], "len": 5, "lerer": 45, "less": [1, 14], "let": [4, 5], "level": [2, 7, 12, 20, 40, 41, 44, 49], "leverag": [3, 14, 39, 47], "librari": [39, 45, 46], "licens": [1, 3, 4, 5], "lightspe": 45, "lihi": 45, "like": 48, "limit": [39, 49], "lin": 45, "linalg": [14, 22], "line": 7, "linear": [0, 15, 17, 23, 24, 28, 31, 32, 33], "link": [7, 46], "list": 46, "liu": 45, "lkopf": 45, "load": 7, "load_digit": 4, "load_numpy_from_url": 3, "local": [1, 46], "log": [9, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 42, 44, 49], "logaffin": [20, 21, 26, 27, 30, 49], "loss": [0, 3, 4, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42, 45, 49], "loss_fn": [3, 4, 9, 42], "low": [5, 39, 49], "lower": [36, 37], "lowrank": 14, "lowrank_nit": 14, "lowrank_q": 14, "lr": [3, 4, 9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42], "lr_affinity_in": 32, "luca": 45, "m": [1, 36, 37, 39, 46], "m18": 33, "m_sampl": 38, "maaten": [12, 31, 45], "machin": [45, 46], "magic": [18, 39, 48], "mai": [14, 30, 32], "main": 46, "maintain": 46, "make": [39, 46], "make_swiss_rol": 5, "manhattan": [15, 17, 23, 31, 32, 33], "mani": 49, "manifold": [0, 5, 33, 45], "manor": [26, 45], "manual_se": 1, "marco": 45, "margin": 49, "markov": 49, "mass": [12, 30, 49], "massa": 45, "massia": 39, "match": [3, 9, 39, 42, 49], "matcher": [6, 7, 9, 10, 12, 19, 20, 23, 24, 28, 31, 33, 40, 41, 42, 43, 44, 47, 49], "mathbb": [11, 12, 18, 26, 27, 30, 34, 36, 37], "mathbf": [9, 11, 12, 13, 15, 17, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 42, 49], "mathcal": [9, 11, 19, 24, 27, 28, 42, 49], "mathemat": [45, 49], "mathieu": 45, "mathop": [11, 12, 27, 30], "mathrm": [11, 12, 19, 24, 27, 28, 30, 34, 49], "mathurin": 39, "matplotlib": [1, 3, 4, 5], "matric": [9, 11, 27, 41, 42, 44, 45], "matrix": [0, 1, 9, 10, 11, 12, 13, 16, 18, 19, 20, 21, 24, 25, 26, 27, 28, 29, 30, 34, 35, 38, 39, 41, 42, 43, 48, 49], "matsumoto": 45, "matur": 46, "max": 1, "max_it": [3, 4, 5, 9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "max_iter_affin": [15, 17, 23, 31, 33], "max_iter_affinity_in": 32, "maxim": 47, "maximum": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "mb": [2, 6, 8, 40], "mcinn": [33, 34, 35, 45, 49], "mean": 4, "mechan": 14, "meet": 46, "mei": 45, "melvil": [33, 45], "mem": [2, 6, 8, 40], "memori": [0, 14, 39, 45, 48], "mention": 46, "merg": 46, "meta": 14, "metadata": 14, "metadata_rout": 14, "method": [0, 4, 9, 10, 14, 19, 22, 24, 28, 36, 37, 39, 42, 43, 46, 47, 48], "method_nam": 3, "metric": [11, 12, 13, 15, 17, 18, 20, 21, 23, 26, 27, 29, 30, 31, 32, 33, 34, 35, 38, 41, 44], "metric_in": [15, 17, 23, 31, 32, 33], "metric_out": [15, 17, 23, 31, 32, 33], "mhm18": 45, "mi": 45, "michel": 45, "mika": 45, "min": [11, 12, 27, 30], "min_": [9, 19, 24, 28, 42, 49], "min_batch_s": 14, "min_dist": [33, 35], "min_j": 34, "ming": 45, "minibatch": 14, "minim": [0, 3, 12, 49], "minimum": [14, 33, 35], "minut": [1, 3, 4, 5], "misc": 39, "mistakenli": 5, "mmd": 45, "mnist": 39, "mnist_784": 39, "mode": 34, "model": [0, 4, 9, 10, 14, 16, 22, 42, 43], "modul": [16, 22, 39, 47], "modular": [0, 39, 47], "moon": 45, "mordant": 45, "more": [12, 30, 32, 49], "most": 49, "mrg": 46, "mu": 30, "much": 46, "multipl": [46, 48], "n": [3, 8, 11, 12, 14, 15, 17, 18, 26, 27, 30, 33, 36, 37, 39, 49], "n_": [12, 30, 34], "n_class": 4, "n_cluster": 1, "n_compon": [3, 4, 5, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 39, 42, 43], "n_featur": [9, 10, 14, 16, 22, 38, 42, 43], "n_neg": [15, 17, 24, 33], "n_neighbor": [33, 34, 49], "n_sampl": [5, 9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 38, 42, 43], "nadam": [11, 15, 17, 23, 30, 31, 32, 33], "naini": 45, "name": [14, 22], "natalia": 45, "ndarrai": [9, 10, 14, 16, 22, 42, 43], "ne": 49, "nearest": [12, 15, 17, 18, 23, 26, 30, 31, 32, 33, 34, 49], "necessari": 46, "need": 14, "neg": [0, 15, 17, 24, 33, 49], "neighbor": [1, 5, 6, 7, 9, 10, 12, 15, 17, 18, 19, 20, 23, 24, 26, 28, 30, 31, 32, 33, 34, 39, 40, 41, 42, 43, 44, 45, 47, 48], "neighborembed": [28, 32], "neighborhood": 49, "neirest": [18, 26], "neural": [33, 45], "neurip": 33, "new": [0, 14, 39, 47, 49], "nicola": [39, 45], "nikla": 45, "nn": [12, 39, 48], "no_grad": 27, "nodiag": 16, "nois": [2, 5, 7, 12, 15, 20, 33, 40, 41, 44, 49], "none": [1, 3, 9, 10, 11, 14, 15, 17, 18, 19, 22, 23, 24, 26, 28, 31, 32, 33, 35, 36, 37, 38, 42, 43], "nonneg": 45, "norm": [11, 49], "normal": [1, 3, 4, 11, 15, 17, 18, 20, 21, 23, 26, 27, 31, 32, 33, 39, 48, 49], "normalization_dim": [1, 3, 20, 21, 26], "normalizedgaussianaffin": [1, 3, 49], "notabl": 49, "note": [0, 14, 19], "notebook": [1, 3, 4, 5, 7], "notimplementederror": [10, 43], "notion": 0, "now": [1, 4], "np": [3, 9, 10, 14, 16, 22, 42, 43], "nu": [21, 29], "nuanc": 49, "number": [1, 9, 10, 11, 12, 14, 15, 16, 17, 19, 22, 23, 24, 27, 28, 30, 31, 32, 33, 34, 39, 42, 43, 46, 49], "numpi": [0, 3, 39], "numpy_arrai": 3, "object": [0, 1, 4, 9, 14, 16, 19, 24, 28, 32, 39, 42, 48, 49], "observ": [1, 5], "obtain": 4, "occur": 1, "offer": [39, 49], "often": 49, "oi": 45, "olivi": 45, "onc": 46, "one": [12, 32, 49], "ones": [11, 12, 27, 30, 49], "onli": [1, 5, 12, 14, 22, 49], "onto": [10, 11, 14, 16, 22, 27, 43], "open": [39, 46], "oper": [0, 41, 44, 45], "optim": [0, 3, 9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 39, 42, 45, 49], "optimizer_kwarg": [3, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "optimum": 12, "option": [9, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 44], "order": [9, 42], "origin": [14, 46], "ot": 32, "other": [14, 45, 46], "otherwis": 14, "out": [1, 48], "outlin": [46, 49], "output": [0, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 39, 42, 46, 48, 49], "output_dim": [15, 17, 23, 31, 32, 33], "outsid": 49, "overflow": [0, 39, 45], "overridden": [10, 43], "overwritten": 14, "p": [11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33], "p_": [12, 15, 17, 18, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34], "pacif": 45, "pairwis": [0, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 38, 39, 41, 44, 49], "paper": 49, "paralleliz": 49, "param": 3, "paramet": [0, 1, 5, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 41, 42, 43, 44, 49], "parent": 49, "part": 49, "partial": 14, "partial_fit": 14, "particular": 0, "particularli": 49, "pass": [14, 46], "paszk": [45, 49], "pattabiraman": 45, "paul": 45, "pca": [5, 6, 7, 9, 10, 14, 15, 17, 19, 23, 24, 25, 28, 31, 32, 33, 39, 40, 41, 42, 43, 47, 48, 49], "pedregosa": [45, 49], "per": 1, "perform": [0, 4, 9, 10, 11, 12, 14, 16, 22, 27, 36, 37, 42, 43, 45, 49], "perona": [26, 45], "perplex": [1, 12, 15, 17, 23, 30, 31, 32, 34, 39, 49], "perplexity_valu": 5, "peter": 45, "peyr": 45, "pgm": 45, "phase": [15, 17, 19, 23, 24, 28, 31, 32, 33], "pick": 46, "pietro": 45, "pip": [7, 39, 47], "pipelin": [14, 47, 49], "place": 46, "plai": 49, "plan": 46, "pleas": [14, 39, 46], "plot": 1, "plot_affinity_graph": 1, "plt": [1, 3, 4, 5], "pmlr": 45, "point": [0, 1, 5, 12, 15, 17, 18, 24, 26, 27, 33, 35, 38, 46, 49], "polytechniqu": 5, "pooja": 45, "popular": 49, "posit": [36, 37, 49], "possibl": [0, 46], "power": 47, "pr": [46, 48], "practic": [30, 32], "pre": 48, "precis": [11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 34, 36, 37], "precomput": [9, 10, 42, 43], "prefer": 46, "prefix": 46, "prepar": 46, "preprint": [33, 45], "present": [0, 49], "preserv": [39, 49], "prettenhof": 45, "prevent": [46, 49], "previous": 14, "princip": [14, 16, 22, 49], "print": [3, 10, 16, 22, 36, 37, 41, 43, 44, 49], "prior": 46, "problem": [0, 9, 11, 12, 19, 24, 27, 28, 30, 39, 42, 49], "problemat": 49, "proceed": 45, "process": [9, 14, 19, 24, 28, 33, 42, 45], "product": [4, 25, 39, 48, 49], "progress": 46, "proj": [11, 27], "project": [5, 10, 11, 14, 16, 22, 27, 33, 39, 43, 45, 46, 48], "properti": 1, "propos": 49, "provid": [0, 9, 14, 35, 39, 42, 47, 49], "public": 46, "pull": [46, 48], "purpos": 46, "pvg": 45, "py": [1, 2, 3, 4, 5, 6, 40], "pykeop": [38, 39, 49], "pypi": 39, "pyplot": [1, 3, 4, 5], "pytest": 46, "python": [1, 3, 4, 5, 7, 39, 45, 46], "pytorch": [0, 9, 14, 15, 17, 23, 30, 31, 32, 33, 39, 42, 45], "q": [15, 17, 19, 23, 24, 27, 28, 31, 32, 33], "q_": [15, 17, 19, 23, 24, 27, 28, 31, 32, 33], "qiaozhu": 45, "quad": [11, 12, 27, 30, 34, 49], "quadrat": [11, 39, 48, 49], "qualiti": 48, "question": 39, "r": [11, 12, 18, 26, 27, 30, 34, 36, 37, 39, 45], "rais": [10, 14, 43, 48], "randn": 49, "random": [9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43, 49], "random_st": [5, 9, 10, 14, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43], "rang": [1, 5, 39, 49], "rangl": [11, 12, 27, 30, 49], "rate": [9, 11, 15, 17, 19, 23, 24, 28, 30, 31, 32, 33, 42], "read": 3, "readi": 46, "receiv": 49, "recent": 49, "recommend": [3, 12], "recov": [45, 49], "reduc": [3, 24], "reduct": [9, 10, 14, 33, 42, 43, 45, 47], "refer": [33, 39, 49], "regardless": 1, "region": 1, "regular": [11, 27], "regularis": 45, "relat": [12, 30], "relev": 14, "reli": [0, 39, 49], "remark": 1, "remedi": 1, "remi": 5, "remov": 16, "rep": [19, 24, 28, 49], "replac": 49, "repositori": 46, "repres": 49, "represent": [39, 49], "reproduc": [3, 9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43, 47], "repuls": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "request": [3, 14, 46, 48], "requir": [32, 39, 46, 49], "research": [39, 45], "resolv": 46, "resort": 32, "respect": [1, 49], "respons": 3, "rest": 47, "restrict": [36, 37], "result": [11, 15, 17, 23, 27, 30, 31, 32, 33, 46, 49], "retain": 14, "return": [3, 9, 14, 16, 22, 36, 37, 38, 42], "review": 46, "rho_i": 34, "richard": 45, "right": [11, 18, 21, 26, 27, 29, 30, 35], "role": 49, "roll": [6, 7, 9, 10, 19, 28, 31, 40, 42, 43, 47], "ron": 45, "root": [0, 12, 15, 17, 23, 31, 34, 36, 37, 46], "roshan": 45, "rotat": 4, "rout": 14, "row": [3, 12, 18, 30, 49], "rowei": [12, 23, 45], "rsinghlab": 3, "run": [0, 1, 4, 5, 7, 46, 47], "r\u00e9mi": [5, 39], "sam": 45, "same": [0, 1, 3, 4], "sampl": [0, 11, 12, 14, 15, 17, 18, 23, 24, 26, 27, 30, 31, 32, 33, 49], "sampledneighborembed": [0, 15, 17, 33], "satur": 12, "scalar": [4, 25, 39, 48, 49], "scalarproductaffin": [0, 4, 49], "scale": [0, 9, 12, 15, 17, 19, 23, 24, 28, 31, 32, 33, 41, 42, 44, 45, 49], "scatter": [1, 3, 4, 5], "sch": 45, "schedul": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 42], "scheduler_kwarg": [9, 17, 19, 23, 24, 28, 31, 32, 33, 42], "schiebing": 45, "scikit": [14, 39, 45], "score": [39, 48], "scot": 3, "script": [1, 3, 4, 5], "se": 30, "sea": 30, "seamlessli": [0, 39, 47], "search": [0, 12, 15, 17, 23, 31, 34, 36, 37], "sebastian": [33, 45], "second": [1, 3, 4, 5, 38], "section": 0, "see": [4, 14, 49], "seed": [9, 10, 15, 16, 17, 19, 22, 23, 24, 28, 31, 32, 33, 42, 43], "seek": 46, "select": [12, 15, 17, 23, 30, 31, 32, 33], "self": [9, 14, 16, 22, 26, 39, 42, 45, 48], "semi": 49, "sequenc": 14, "serv": [46, 49], "set": [0, 1, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 26, 27, 29, 30, 32, 33, 34, 35, 41, 47, 49], "set_config": 14, "set_fit_request": 14, "set_partial_fit_request": 14, "set_titl": [3, 5], "set_xtick": 3, "set_ytick": 3, "sgd": [11, 15, 17, 23, 30, 31, 32, 33], "shannon": [12, 27, 30, 49], "shape": [9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 36, 37, 38, 42, 43, 49], "share": [47, 49], "sharma": 45, "short": 5, "should": [10, 14, 24, 28, 36, 37, 43, 46, 49], "show": [1, 4, 5], "shun": 45, "sigma": [1, 13, 18, 20, 26, 34], "sigma_i": [18, 26], "sigma_j": 26, "signific": 46, "significantli": [3, 5, 15, 17, 23, 31, 32], "silhouett": [39, 48], "similar": [0, 34, 39, 49], "simpli": 12, "simplifi": 39, "simul": 1, "sinc": 12, "singl": [39, 45], "singular": 4, "sinkhorn": [27, 45], "sinkhornaffin": [32, 49], "size": [1, 14, 36, 37, 49], "sk67": 45, "sklearn": [4, 5, 14, 39, 49], "slice": 14, "small": [12, 49], "smallest": 12, "snare_rna": 3, "snareseq_typ": 3, "sne": [3, 31, 39, 45, 48, 49], "sne_affinity_match": 3, "snekhorn": [12, 39, 45, 48, 49], "soft": [12, 49], "sole": [39, 49], "solut": [4, 30, 49], "solv": [0, 9, 11, 12, 19, 24, 27, 28, 32, 39, 42, 48, 49], "solver": 0, "some": 14, "someth": 46, "soon": 46, "sourc": [1, 3, 4, 5, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46], "space": [0, 5, 9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 35, 42, 49], "spars": [1, 28, 45, 49], "sparselogaffin": [12, 24, 28, 34], "sparseneighborembed": [0, 23, 24, 31], "sparsiti": [0, 1, 3, 12, 24, 34, 39], "speak": 0, "specif": [0, 3], "specifi": [20, 21], "spectral": [4, 39, 45, 48], "speed": 39, "sphinx": [1, 3, 4, 5, 7], "sphinxcontrib": 48, "split": 3, "spread": [12, 33, 35, 49], "sqeuclidean": [11, 12, 13, 15, 17, 18, 20, 21, 23, 26, 27, 29, 30, 31, 32, 33, 34, 35, 38, 41, 44], "sqrt": 49, "squar": [4, 30, 32, 49], "square_loss": [4, 9, 42], "stabl": [30, 32], "standard": [1, 46], "star": [11, 12, 27, 30, 49], "star_": 30, "star_i": [11, 27, 30, 34], "star_j": [11, 27, 30], "start": 46, "static": 14, "statist": 45, "statu": 46, "std": 1, "stephen": 45, "stochast": [11, 23, 24, 27, 31, 39, 45, 48, 49], "stop": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 36, 37, 42], "store": [11, 15, 17, 23, 27, 30, 31, 32, 33, 49], "str": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 37, 38, 41, 42, 43, 44], "straightforward": 49, "strategi": [0, 39], "student": [11, 21, 27, 29, 33, 35, 39, 48, 49], "studentaffin": [17, 31, 49], "studi": 33, "style": [45, 46, 48], "sub": 14, "subclass": [10, 43], "submit": 46, "subplot": [1, 5], "subproblem": 12, "subspac": 14, "suggest": 39, "sum": 49, "sum_": [12, 15, 17, 18, 19, 23, 24, 27, 28, 30, 31, 32, 33, 49], "sum_i": [15, 23, 49], "sum_j": [34, 49], "summar": 49, "support": [0, 39], "sure": 46, "svd": [6, 7, 9, 10, 14, 22, 25, 40, 41, 42, 43, 47], "svd_driver": [14, 22], "svd_lowrank": 14, "swiss": [6, 7, 9, 10, 19, 28, 31, 40, 42, 43, 47], "switch": 48, "symbol": 39, "symmetr": [11, 12, 27, 30, 32, 39, 45, 48, 49], "symmetri": [30, 49], "symmetric_affin": 32, "symmetricentropicaffin": [12, 32, 49], "system": [33, 45], "t": [3, 5, 12, 18, 21, 29, 30, 31, 33, 35, 39, 45, 48], "take": [3, 49], "taken": [46, 47, 49], "tang": [17, 45], "target": 4, "task": 46, "techniqu": 39, "templat": 48, "tensor": [0, 1, 9, 10, 11, 14, 15, 16, 17, 22, 23, 31, 32, 33, 36, 37, 38, 39, 42, 43], "term": [0, 15, 17, 19, 23, 24, 28, 31, 32, 33, 49], "termin": 47, "test": 46, "tetsuya": 45, "text": [5, 11, 12, 27, 30, 34, 46, 49], "text2d": 5, "th": [18, 26], "than": [14, 49], "thank": [39, 46], "thei": [0, 49], "them": [3, 7], "therebi": 49, "therefor": 3, "thi": [0, 1, 10, 11, 14, 22, 24, 27, 28, 32, 39, 43, 46, 49], "thibault": 45, "thirion": 45, "those": [0, 49], "three": 1, "threshold": [11, 12, 15, 17, 23, 27, 30, 31, 32, 33, 34, 36, 37], "through": 39, "thu": [1, 12, 49], "tight_layout": 3, "time": [0, 1, 3, 4, 5, 11, 12, 27, 30], "titl": [1, 4, 5, 39, 46], "titouan": [3, 39, 45], "tlzm16": 45, "toi": 1, "tol": [9, 11, 12, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 34, 36, 37, 42], "tol_affin": [15, 17, 23, 31, 33], "tol_affinity_in": 32, "toler": [9, 19, 24, 28, 42], "tolog": [9, 11, 15, 17, 19, 23, 24, 27, 28, 30, 31, 32, 33, 42], "too": [5, 12, 14, 32], "toolbox": 39, "top": [7, 11, 12, 25, 27, 30, 49], "torch": [1, 9, 10, 11, 14, 15, 16, 17, 22, 23, 27, 31, 32, 33, 36, 37, 38, 42, 43], "torchdr": [1, 4, 5, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 46, 49], "total": [1, 2, 3, 4, 5, 6, 8, 40], "track": [27, 46], "tracker": [39, 46], "train": 14, "transax": 5, "transform": [5, 9, 10, 14, 16, 22, 39, 42, 43], "transformermixin": 49, "transit": 49, "transport": [11, 12, 27, 32, 45], "transpos": [12, 18], "trevor": 45, "triplet": 49, "trouv": 45, "true": [0, 9, 10, 11, 12, 13, 14, 18, 19, 20, 21, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 45, 49], "try": 46, "tsne": [0, 1, 3, 6, 7, 9, 10, 12, 19, 28, 30, 39, 40, 42, 43, 47, 49], "tsnekhorn": [12, 49], "tune": [26, 39, 45, 48], "tupl": [20, 21, 26], "twenti": 45, "two": [0, 3, 9, 38, 42, 49], "two_sne_dict": 3, "txt": 3, "type": [9, 14, 16, 22, 36, 37, 39, 42, 46], "typic": [41, 44, 49], "u": 4, "ultim": 49, "umap": [3, 34, 35, 39, 45, 48, 49], "umapaffinityin": [33, 49], "umapaffinityout": [33, 49], "unchang": 14, "under": 49, "uniform": [33, 45], "uniformli": 49, "unit": 12, "unlik": 30, "unnorm": 16, "unnormalizedaffin": [24, 25, 28, 35], "unnormalizedlogaffin": [13, 29], "unrol": 32, "unstabl": 32, "up": 4, "updat": [14, 30, 32, 46], "upper": [36, 37], "url": [3, 39], "url_i": 3, "url_x": 3, "urllib": 3, "urlopen": 3, "us": [0, 1, 4, 11, 13, 14, 15, 16, 17, 18, 21, 26, 27, 29, 30, 32, 34, 35, 38, 39, 45, 46], "usag": 3, "user": [14, 39], "usual": [39, 49], "utf": 3, "util": [14, 49], "v": 49, "valid": 14, "valu": [4, 12, 15, 17, 23, 30, 31, 32], "van": [1, 3, 4, 5, 12, 18, 30, 31, 32, 39, 45, 49], "vanassel2024torchdr": 39, "vanasselhugu": [1, 3, 4, 5], "varepsilon": [11, 12, 27, 30], "varepsilon_i": 12, "vari": [2, 7, 12, 20, 40, 41, 44, 49], "variabl": [11, 12, 27, 30, 32, 34], "varieti": 0, "variou": [0, 1, 39, 47], "varoquaux": 45, "vavfc24": 45, "vayer": [3, 39, 45], "vdmh08": 45, "vdsn": 45, "vector": [11, 12, 27, 30], "verbos": [1, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 41, 42, 43, 44], "veri": 1, "version": [12, 14, 30, 39], "via": [0, 3, 6, 7, 9, 10, 22, 25, 27, 34, 39, 40, 41, 42, 43, 47, 48, 49], "vialard": 45, "view": [45, 49], "view_init": 5, "viewabl": 46, "vincent": [39, 45], "visual": 45, "wai": [0, 11, 27, 39, 46, 47], "we": [1, 3, 4, 5, 46, 49], "web": [45, 46], "weight": 49, "weiss": 45, "welcom": 46, "well": [0, 4, 39, 49], "what": 49, "when": [3, 5, 14, 24, 28, 32, 46, 48, 49], "where": [9, 11, 12, 13, 15, 17, 19, 20, 21, 24, 25, 27, 28, 29, 30, 33, 34, 35, 39, 42, 46, 49], "whether": [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 41, 42, 43, 44, 49], "which": [10, 11, 12, 14, 15, 16, 17, 19, 20, 21, 22, 23, 26, 27, 30, 31, 32, 33, 36, 37, 39, 43, 49], "while": [0, 12, 47, 49], "wide": [39, 45, 49], "wip": 46, "wise": [12, 18, 26, 30], "with_grad": 27, "within": 49, "without": [0, 39, 45, 46, 49], "work": [0, 14, 22, 46, 47], "workflow": 46, "world": 45, "write": [11, 27], "written": 39, "x": [1, 3, 4, 5, 9, 10, 14, 16, 18, 22, 25, 26, 38, 39, 42, 43, 49], "x1": 1, "x2": 1, "x3": 1, "x_": 39, "x_embed": 5, "x_new": [14, 16, 22], "xavier": 45, "xi": [12, 30], "y": [3, 9, 10, 38, 42, 43], "year": 39, "yield": 14, "yim": 45, "you": [0, 7, 14, 39, 46, 47], "your": [0, 39, 46, 47], "z": [9, 15, 17, 19, 23, 24, 28, 31, 32, 33, 39, 42, 49], "z_am": 4, "z_gpu": 39, "z_svd": 4, "zelnik": [26, 45], "zeme": 45, "zero": [13, 16, 18, 20, 21, 26, 29, 34, 35, 41], "zero_diag": [1, 11, 12, 13, 18, 20, 21, 26, 27, 29, 30, 34, 35, 41, 44], "zhang": [11, 45, 49], "zip": [1, 3, 4, 5, 7], "zmms23": 45, "zmp04": 45, "\u00e7": 45, "\u00e9": 45, "\u00eb": 45, "\u00f6": 45}, "titles": ["API and Modules", "Entropic Affinities can adapt to varying noise levels", "Computation times", "Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation", "PCA via SVD and via AffinityMatcher", "TSNE embedding of the swiss roll dataset", "Computation times", "Gallery", "Computation times", "AffinityMatcher", "DRModule", "DoublyStochasticQuadraticAffinity", "EntropicAffinity", "GaussianAffinity", "IncrementalPCA", "InfoTSNE", "KernelPCA", "LargeVis", "MAGICAffinity", "NeighborEmbedding", "NormalizedGaussianAffinity", "NormalizedStudentAffinity", "PCA", "SNE", "SampledNeighborEmbedding", "ScalarProductAffinity", "SelfTuningAffinity", "SinkhornAffinity", "SparseNeighborEmbedding", "StudentAffinity", "SymmetricEntropicAffinity", "TSNE", "TSNEkhorn", "UMAP", "UMAPAffinityIn", "UMAPAffinityOut", "binary_search", "false_position", "pairwise_distances", "Torch Dimensionality Reduction", "Computation times", "Affinity", "AffinityMatcher", "DRModule", "LogAffinity", "Bibliography", "How to Contribute", "Quick Start Guide", "Releases", "User Guide"], "titleterms": {"0": 48, "09": 48, "1": 48, "17": 48, "2024": 48, "On": 3, "adapt": 1, "advanc": 0, "affin": [0, 1, 3, 7, 39, 41, 49], "affinitymatch": [3, 4, 9, 42, 47], "algorithm": 39, "api": [0, 3], "automat": 49, "avoid": 49, "bandwidth": 1, "base": [0, 49], "basic": 7, "bibliographi": 45, "binary_search": 36, "can": 1, "cell": 3, "citat": 39, "class": 0, "compat": 0, "comput": [2, 5, 6, 8, 40], "constant": 1, "content": 49, "contribut": 46, "contributor": 46, "dataset": [3, 5], "date": 39, "differ": 3, "differenti": 49, "dimension": [0, 39, 49], "directli": 3, "distanc": 0, "doublystochasticquadraticaffin": 11, "drmodul": [10, 43], "effici": 3, "embed": [0, 3, 5, 49], "entrop": [0, 1, 49], "entropicaffin": [12, 49], "equival": 3, "estim": 0, "evalu": 39, "exampl": [9, 10, 12, 19, 20, 22, 23, 24, 25, 28, 31, 33, 41, 42, 43, 44, 47, 49], "express": 3, "false_posit": 37, "featur": 39, "find": 39, "formul": [3, 49], "galleri": 7, "gaussian": 1, "gaussianaffin": 13, "gene": 3, "gener": 49, "genom": 3, "get": 39, "gpu": 49, "guid": [47, 49], "help": 39, "how": 46, "imag": [4, 5], "impact": 5, "implement": 39, "incrementalpca": 14, "infotsn": 15, "instal": [39, 47], "keop": 49, "kernelpca": 16, "knn": 0, "label": 3, "largevi": 17, "lazi": 49, "level": 1, "load": [3, 4, 5], "logaffin": 44, "magicaffin": 18, "matcher": 3, "memori": 49, "method": [3, 49], "metric": 39, "modul": [0, 49], "neighbor": [0, 3, 49], "neighborembed": 19, "new": 46, "nois": 1, "normal": 0, "normalis": 1, "normalizedgaussianaffin": 20, "normalizedstudentaffin": 21, "other": 49, "overflow": 49, "overview": 49, "pairwise_dist": 38, "pca": [4, 22], "perplex": 5, "plot": [3, 5], "project": 49, "quadrat": 0, "quick": 47, "rather": 3, "reduct": [0, 39, 49], "releas": 48, "roll": 5, "row": 1, "run": 3, "sampledneighborembed": 24, "scalarproductaffin": 25, "see": 5, "selftuningaffin": 26, "seq": 3, "simpl": 0, "sinkhornaffin": 27, "sklearn": 0, "snare": 3, "sne": 23, "sparseneighborembed": 28, "spectral": [0, 49], "spotlight": 49, "start": [39, 47], "structur": 49, "studentaffin": 29, "support": 49, "svd": 4, "swiss": 5, "symbol": 49, "symmetricentropicaffin": 30, "tabl": 49, "tensor": 49, "than": 3, "time": [2, 6, 8, 40], "tip": 46, "toi": [4, 5], "torch": [39, 49], "torchdr": [0, 3, 47], "tsne": [5, 31], "tsnekhorn": 32, "type": 3, "umap": [0, 33], "umapaffinityin": 34, "umapaffinityout": 35, "us": [3, 9, 10, 12, 19, 20, 22, 23, 24, 25, 28, 31, 33, 41, 42, 43, 44, 47, 49], "user": 49, "util": 0, "vari": 1, "variou": 49, "version": 48, "via": 4, "when": 47}})
\ No newline at end of file
diff --git a/dev/sg_execution_times.html b/dev/sg_execution_times.html
index 7d4d3e0..2c7a54a 100644
--- a/dev/sg_execution_times.html
+++ b/dev/sg_execution_times.html
@@ -84,7 +84,7 @@
              
   <section id="computation-times">
 <span id="sphx-glr-sg-execution-times"></span><h1>Computation times<a class="headerlink" href="#computation-times" title="Link to this heading"></a></h1>
-<p><strong>02:12.130</strong> total execution time for 4 files <strong>from all galleries</strong>:</p>
+<p><strong>02:38.054</strong> total execution time for 4 files <strong>from all galleries</strong>:</p>
 <div class="docutils container">
 <style scoped>
 <link href="https://cdnjs.cloudflare.com/ajax/libs/twitter-bootstrap/5.3.0/css/bootstrap.min.css" rel="stylesheet" />
@@ -105,20 +105,20 @@
 </tr>
 </thead>
 <tbody>
-<tr class="row-even"><td><p><a class="reference internal" href="auto_examples/basics/demo_tsne_swiss_roll.html#sphx-glr-auto-examples-basics-demo-tsne-swiss-roll-py"><span class="std std-ref">TSNE embedding of the swiss roll dataset</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/basics/demo_tsne_swiss_roll.py</span></code>)</p></td>
-<td><p>01:07.864</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="auto_examples/basics/demo_ne_methods_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-ne-methods-affinity-matcher-py"><span class="std std-ref">Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/basics/demo_ne_methods_affinity_matcher.py</span></code>)</p></td>
+<td><p>01:14.869</p></td>
 <td><p>0.0</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="auto_examples/basics/demo_ne_methods_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-ne-methods-affinity-matcher-py"><span class="std std-ref">Neighbor Embedding on genomics &amp; equivalent affinity matcher formulation</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/basics/demo_ne_methods_affinity_matcher.py</span></code>)</p></td>
-<td><p>00:56.286</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="auto_examples/basics/demo_tsne_swiss_roll.html#sphx-glr-auto-examples-basics-demo-tsne-swiss-roll-py"><span class="std std-ref">TSNE embedding of the swiss roll dataset</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/basics/demo_tsne_swiss_roll.py</span></code>)</p></td>
+<td><p>01:14.668</p></td>
 <td><p>0.0</p></td>
 </tr>
 <tr class="row-even"><td><p><a class="reference internal" href="auto_examples/basics/demo_pca_via_affinity_matcher.html#sphx-glr-auto-examples-basics-demo-pca-via-affinity-matcher-py"><span class="std std-ref">PCA via SVD and via AffinityMatcher</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/basics/demo_pca_via_affinity_matcher.py</span></code>)</p></td>
-<td><p>00:05.006</p></td>
+<td><p>00:05.351</p></td>
 <td><p>0.0</p></td>
 </tr>
 <tr class="row-odd"><td><p><a class="reference internal" href="auto_examples/affinities/demo_ea_adaptivity.html#sphx-glr-auto-examples-affinities-demo-ea-adaptivity-py"><span class="std std-ref">Entropic Affinities can adapt to varying noise levels</span></a> (<code class="docutils literal notranslate"><span class="pre">../../examples/affinities/demo_ea_adaptivity.py</span></code>)</p></td>
-<td><p>00:02.973</p></td>
+<td><p>00:03.166</p></td>
 <td><p>0.0</p></td>
 </tr>
 </tbody>