deploy: cf4c54b

Author: LouisTier

_modules/torchsom/visualization/base.html

@@ -608,6 +608,7 @@ <h1>Source code for torchsom.visualization.base</h1><div class="highlight"><pre>
         <span class="bp">self</span><span class="p">,</span>
         <span class="n">quantization_errors</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">float</span><span class="p">],</span>
         <span class="n">topographic_errors</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">float</span><span class="p">],</span>
+        <span class="n">bmus_data_map</span><span class="p">:</span> <span class="nb">dict</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="nb">int</span><span class="p">],</span> <span class="nb">list</span><span class="p">[</span><span class="nb">int</span><span class="p">]],</span>
         <span class="n">data</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">,</span>
         <span class="n">target</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">,</span>
         <span class="n">component_names</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
@@ -625,6 +626,7 @@ <h1>Source code for torchsom.visualization.base</h1><div class="highlight"><pre>
 <span class="sd">        Args:</span>
 <span class="sd">            quantization_errors (list[float]): List of quantization errors [epochs]</span>
 <span class="sd">            topographic_errors (list[float]): List of topographic errors [epochs]</span>
+<span class="sd">            bmus_data_map (dict[tuple[int, int], list[int]]): Pre-computed BMU to data indices mapping</span>
 <span class="sd">            data (torch.Tensor): Input data tensor [batch_size, n_features]</span>
 <span class="sd">            target (torch.Tensor): Labels tensor for data points [batch_size]</span>
 <span class="sd">            component_names (Optional[list[str]]): Names for each component/feature</span>
@@ -649,15 +651,30 @@ <h1>Source code for torchsom.visualization.base</h1><div class="highlight"><pre>
             <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_hit_map</span><span class="p">(</span><span class="n">data</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">)</span>
         <span class="k">if</span> <span class="n">metric_map</span><span class="p">:</span>
             <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_metric_map</span><span class="p">(</span>
-                <span class="n">data</span><span class="p">,</span> <span class="n">target</span><span class="p">,</span> <span class="n">reduction_parameter</span><span class="o">=</span><span class="s2">&quot;mean&quot;</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span>
+                <span class="n">bmus_data_map</span><span class="o">=</span><span class="n">bmus_data_map</span><span class="p">,</span>
+                <span class="n">data</span><span class="o">=</span><span class="n">data</span><span class="p">,</span>
+                <span class="n">target</span><span class="o">=</span><span class="n">target</span><span class="p">,</span>
+                <span class="n">reduction_parameter</span><span class="o">=</span><span class="s2">&quot;mean&quot;</span><span class="p">,</span>
+                <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">,</span>
             <span class="p">)</span>
             <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_metric_map</span><span class="p">(</span>
-                <span class="n">data</span><span class="p">,</span> <span class="n">target</span><span class="p">,</span> <span class="n">reduction_parameter</span><span class="o">=</span><span class="s2">&quot;std&quot;</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span>
+                <span class="n">bmus_data_map</span><span class="o">=</span><span class="n">bmus_data_map</span><span class="p">,</span>
+                <span class="n">data</span><span class="o">=</span><span class="n">data</span><span class="p">,</span>
+                <span class="n">target</span><span class="o">=</span><span class="n">target</span><span class="p">,</span>
+                <span class="n">reduction_parameter</span><span class="o">=</span><span class="s2">&quot;std&quot;</span><span class="p">,</span>
+                <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">,</span>
             <span class="p">)</span>
         <span class="k">if</span> <span class="n">score_map</span><span class="p">:</span>
-            <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_score_map</span><span class="p">(</span><span class="n">data</span><span class="p">,</span> <span class="n">target</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">)</span>
+            <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_score_map</span><span class="p">(</span>
+                <span class="n">bmus_data_map</span><span class="o">=</span><span class="n">bmus_data_map</span><span class="p">,</span>
+                <span class="n">target</span><span class="o">=</span><span class="n">target</span><span class="p">,</span>
+                <span class="n">total_samples</span><span class="o">=</span><span class="n">data</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span>
+                <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">,</span>
+            <span class="p">)</span>
         <span class="k">if</span> <span class="n">rank_map</span><span class="p">:</span>
-            <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_rank_map</span><span class="p">(</span><span class="n">data</span><span class="p">,</span> <span class="n">target</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span><span class="p">)</span>
+            <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_rank_map</span><span class="p">(</span>
+                <span class="n">bmus_data_map</span><span class="o">=</span><span class="n">bmus_data_map</span><span class="p">,</span> <span class="n">target</span><span class="o">=</span><span class="n">target</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span>
+            <span class="p">)</span>
         <span class="k">if</span> <span class="n">component_planes</span><span class="p">:</span>
             <span class="bp">self</span><span class="o">.</span><span class="n">_visualizer</span><span class="o">.</span><span class="n">plot_component_planes</span><span class="p">(</span>
                 <span class="n">component_names</span><span class="o">=</span><span class="n">component_names</span><span class="p">,</span> <span class="n">save_path</span><span class="o">=</span><span class="n">save_path</span>

api/visualization.html

@@ -440,13 +440,14 @@ <h1>Visualization API<a class="headerlink" href="#visualization-api" title="Link
 </dl>
 <dl class="py method">
 <dt class="sig sig-object py" id="torchsom.visualization.base.SOMVisualizer.plot_all">
-<span class="sig-name descname"><span class="pre">plot_all</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">quantization_errors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">topographic_errors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">data</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">component_names</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">save_path</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">training_errors</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">distance_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hit_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">score_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rank_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">metric_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">component_planes</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/torchsom/visualization/base.html#SOMVisualizer.plot_all"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchsom.visualization.base.SOMVisualizer.plot_all" title="Link to this definition"></a></dt>
+<span class="sig-name descname"><span class="pre">plot_all</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">quantization_errors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">topographic_errors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">bmus_data_map</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">data</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">component_names</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">save_path</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">training_errors</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">distance_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hit_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">score_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rank_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">metric_map</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">component_planes</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/torchsom/visualization/base.html#SOMVisualizer.plot_all"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchsom.visualization.base.SOMVisualizer.plot_all" title="Link to this definition"></a></dt>
 <dd><p>Plot all visualizations using topology-specific visualizer.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
 <li><p><strong>quantization_errors</strong> (<em>list</em><em>[</em><em>float</em><em>]</em>) – List of quantization errors [epochs]</p></li>
 <li><p><strong>topographic_errors</strong> (<em>list</em><em>[</em><em>float</em><em>]</em>) – List of topographic errors [epochs]</p></li>
+<li><p><strong>bmus_data_map</strong> (<em>dict</em><em>[</em><em>tuple</em><em>[</em><em>int</em><em>, </em><em>int</em><em>]</em><em>, </em><em>list</em><em>[</em><em>int</em><em>]</em><em>]</em>) – Pre-computed BMU to data indices mapping</p></li>
 <li><p><strong>data</strong> (<em>torch.Tensor</em>) – Input data tensor [batch_size, n_features]</p></li>
 <li><p><strong>target</strong> (<em>torch.Tensor</em>) – Labels tensor for data points [batch_size]</p></li>
 <li><p><strong>component_names</strong> (<em>Optional</em><em>[</em><em>list</em><em>[</em><em>str</em><em>]</em><em>]</em>) – Names for each component/feature</p></li>