|
80 | 80 | <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> |
81 | 81 | <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> |
82 | 82 | <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> |
83 | 84 | <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> |
84 | 86 | <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> |
85 | 87 | </ul> |
86 | 88 | </li> |
|
89 | 91 | <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> |
90 | 92 | <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> |
91 | 93 | <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> |
92 | 95 | <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> |
93 | 97 | </ul> |
94 | 98 | </li> |
95 | 99 | </ul> |
|
184 | 188 |
|
185 | 189 | <dl class="py class"> |
186 | 190 | <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> |
188 | 192 | <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> |
189 | 193 | <p>Class used to represent the training strategy based on the Pytorch learning framework. |
190 | 194 | It supports different optimization algorithms, schedulers, loss functions and others based on |
|
356 | 360 | </dl> |
357 | 361 | </dd></dl> |
358 | 362 |
|
| 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 | + |
359 | 374 | <dl class="py method"> |
360 | 375 | <dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.train"> |
361 | 376 | <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 | 392 | </dl> |
378 | 393 | </dd></dl> |
379 | 394 |
|
| 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 | + |
380 | 415 | <dl class="py method"> |
381 | 416 | <dt class="sig sig-object py" id="pynever.strategies.training.PytorchTraining.pytorch_training"> |
382 | 417 | <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 | 436 |
|
402 | 437 | <dl class="py class"> |
403 | 438 | <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> |
405 | 440 | <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> |
406 | 441 | <p>Class used to represent the testing strategy based on the Pytorch learning framework. |
407 | 442 | It supports different metrics measure for the correctness of the neural network.</p> |
|
451 | 486 | </dl> |
452 | 487 | </dd></dl> |
453 | 488 |
|
| 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 | + |
454 | 500 | <dl class="py method"> |
455 | 501 | <dt class="sig sig-object py" id="pynever.strategies.training.PytorchTesting.test"> |
456 | 502 | <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 | 517 | </dl> |
472 | 518 | </dd></dl> |
473 | 519 |
|
| 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 | + |
474 | 540 | </dd></dl> |
475 | 541 |
|
476 | 542 | </section> |
|
0 commit comments