Skip to content

Commit 6430e90

Browse files
author
github-actions
committed
Update API docs from pynever
1 parent f8f8ac3 commit 6430e90

5 files changed

Lines changed: 102 additions & 12 deletions

File tree

pynever/API/2_Training.html

Lines changed: 68 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,9 @@
8080
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.train_patience"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.train_patience</span></code></a></li>
8181
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.checkpoints_root"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.checkpoints_root</span></code></a></li>
8282
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.verbose_rate"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.verbose_rate</span></code></a></li>
83+
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.task"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.task</span></code></a></li>
8384
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.train"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.train()</span></code></a></li>
85+
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.compute_loss"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.compute_loss()</span></code></a></li>
8486
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTraining.pytorch_training"><code class="docutils literal notranslate"><span class="pre">PytorchTraining.pytorch_training()</span></code></a></li>
8587
</ul>
8688
</li>
@@ -89,7 +91,9 @@
8991
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.metric_params"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.metric_params</span></code></a></li>
9092
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.test_batch_size"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.test_batch_size</span></code></a></li>
9193
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.save_results"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.save_results</span></code></a></li>
94+
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.task"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.task</span></code></a></li>
9295
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.test"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.test()</span></code></a></li>
96+
<li class="toctree-l3"><a class="reference internal" href="#pynever.strategies.training.PytorchTesting.compute_metric"><code class="docutils literal notranslate"><span class="pre">PytorchTesting.compute_metric()</span></code></a></li>
9397
</ul>
9498
</li>
9599
</ul>
@@ -184,7 +188,7 @@
184188

185189
<dl class="py class">
186190
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining">
187-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">pynever.strategies.training.</span></span><span class="sig-name descname"><span class="pre">PytorchTraining</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">optimizer_con</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">opt_params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">loss_function</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n_epochs</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">validation_percentage</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">validation_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">r_split</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">scheduler_con</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">sch_params</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">precision_metric</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">network_transform</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">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cpu'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_patience</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">checkpoints_root</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">verbose_rate</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining" title="Link to this definition"></a></dt>
191+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">pynever.strategies.training.</span></span><span class="sig-name descname"><span class="pre">PytorchTraining</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">optimizer_con</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">opt_params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">loss_function</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n_epochs</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">validation_percentage</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">validation_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">r_split</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">scheduler_con</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">sch_params</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">precision_metric</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">network_transform</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">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cpu'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_patience</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">checkpoints_root</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">verbose_rate</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">task</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'classification'</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining" title="Link to this definition"></a></dt>
188192
<dd><p>Bases: <a class="reference internal" href="#pynever.strategies.training.TrainingStrategy" title="pynever.strategies.training.TrainingStrategy"><code class="xref py py-class docutils literal notranslate"><span class="pre">TrainingStrategy</span></code></a></p>
189193
<p>Class used to represent the training strategy based on the Pytorch learning framework.
190194
It supports different optimization algorithms, schedulers, loss functions and others based on
@@ -356,6 +360,17 @@
356360
</dl>
357361
</dd></dl>
358362

363+
<dl class="py attribute">
364+
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.task">
365+
<span class="sig-name descname"><span class="pre">task</span></span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining.task" title="Link to this definition"></a></dt>
366+
<dd><p>Which task the training procedure is expecting between classification and regression. (default: classification)</p>
367+
<dl class="field-list simple">
368+
<dt class="field-odd">Type<span class="colon">:</span></dt>
369+
<dd class="field-odd"><p>str, Optional</p>
370+
</dd>
371+
</dl>
372+
</dd></dl>
373+
359374
<dl class="py method">
360375
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.train">
361376
<span class="sig-name descname"><span class="pre">train</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">network</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dataset</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining.train" title="Link to this definition"></a></dt>
@@ -377,6 +392,26 @@
377392
</dl>
378393
</dd></dl>
379394

395+
<dl class="py method">
396+
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.compute_loss">
397+
<span class="sig-name descname"><span class="pre">compute_loss</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">output</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining.compute_loss" title="Link to this definition"></a></dt>
398+
<dd><p>Compute the selected loss function for the given output and target.</p>
399+
<dl class="field-list simple">
400+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
401+
<dd class="field-odd"><ul class="simple">
402+
<li><p><strong>output</strong> (<em>torch.Tensor</em>) – Output of the neural network</p></li>
403+
<li><p><strong>target</strong> (<em>torch.Tensor</em>) – Target output to match</p></li>
404+
</ul>
405+
</dd>
406+
<dt class="field-even">Returns<span class="colon">:</span></dt>
407+
<dd class="field-even"><p>The computed loss</p>
408+
</dd>
409+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
410+
<dd class="field-odd"><p>torch.Tensor</p>
411+
</dd>
412+
</dl>
413+
</dd></dl>
414+
380415
<dl class="py method">
381416
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.pytorch_training">
382417
<span class="sig-name descname"><span class="pre">pytorch_training</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">net</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dataset</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTraining.pytorch_training" title="Link to this definition"></a></dt>
@@ -401,7 +436,7 @@
401436

402437
<dl class="py class">
403438
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTesting">
404-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">pynever.strategies.training.</span></span><span class="sig-name descname"><span class="pre">PytorchTesting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">metric</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">metric_params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">test_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cpu'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">save_results</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTesting" title="Link to this definition"></a></dt>
439+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">pynever.strategies.training.</span></span><span class="sig-name descname"><span class="pre">PytorchTesting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">metric</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">metric_params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">test_batch_size</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cpu'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">save_results</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">task</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'classification'</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTesting" title="Link to this definition"></a></dt>
405440
<dd><p>Bases: <a class="reference internal" href="#pynever.strategies.training.TestingStrategy" title="pynever.strategies.training.TestingStrategy"><code class="xref py py-class docutils literal notranslate"><span class="pre">TestingStrategy</span></code></a></p>
406441
<p>Class used to represent the testing strategy based on the Pytorch learning framework.
407442
It supports different metrics measure for the correctness of the neural network.</p>
@@ -451,6 +486,17 @@
451486
</dl>
452487
</dd></dl>
453488

489+
<dl class="py attribute">
490+
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTesting.task">
491+
<span class="sig-name descname"><span class="pre">task</span></span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTesting.task" title="Link to this definition"></a></dt>
492+
<dd><p>Which task the training procedure is expecting between classification and regression. (default: classification)</p>
493+
<dl class="field-list simple">
494+
<dt class="field-odd">Type<span class="colon">:</span></dt>
495+
<dd class="field-odd"><p>str, Optional</p>
496+
</dd>
497+
</dl>
498+
</dd></dl>
499+
454500
<dl class="py method">
455501
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTesting.test">
456502
<span class="sig-name descname"><span class="pre">test</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">network</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dataset</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTesting.test" title="Link to this definition"></a></dt>
@@ -471,6 +517,26 @@
471517
</dl>
472518
</dd></dl>
473519

520+
<dl class="py method">
521+
<dt class="sig sig-object py" id="pynever.strategies.training.PytorchTesting.compute_metric">
522+
<span class="sig-name descname"><span class="pre">compute_metric</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">output</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/NeverTools/pynever/blob/main/pynever/strategies/training.py"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#pynever.strategies.training.PytorchTesting.compute_metric" title="Link to this definition"></a></dt>
523+
<dd><p>Compute the selected metric for the given output and target.</p>
524+
<dl class="field-list simple">
525+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
526+
<dd class="field-odd"><ul class="simple">
527+
<li><p><strong>output</strong> (<em>torch.Tensor</em>) – Output of the neural network</p></li>
528+
<li><p><strong>target</strong> (<em>torch.Tensor</em>) – Target output to match</p></li>
529+
</ul>
530+
</dd>
531+
<dt class="field-even">Returns<span class="colon">:</span></dt>
532+
<dd class="field-even"><p>The computed loss</p>
533+
</dd>
534+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
535+
<dd class="field-odd"><p>torch.Tensor</p>
536+
</dd>
537+
</dl>
538+
</dd></dl>
539+
474540
</dd></dl>
475541

476542
</section>

0 commit comments

Comments
 (0)