:orphan:

pytorch interface
=================

Here is a set of examples describing how to use WDRO for custom models via pytorch modules.



.. raw:: html

    <div class="sphx-glr-thumbnails">

.. thumbnail-parent-div-open

.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="This example solves a simple binary classification problem using a basic neural network with 2 layers.">

.. only:: html

  .. image:: /examples/Custom/images/thumb/sphx_glr_neural_net_thumb.png
    :alt:

  :ref:`sphx_glr_examples_Custom_neural_net.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">Simple Neural Network</div>
    </div>


.. raw:: html

    <div class="sphx-glr-thumbcontainer" tooltip="Polynomial regression is a simple 1D regression. The samples are of the form \xi = (x,y) \in \mathbb{R}\times\mathbb{R} and the sought predictor is of the form  f(x) = \sum_{i=0}^d a_i x^i where (a_0,..,a_d) are the d+1 coefficients to lean.">

.. only:: html

  .. image:: /examples/Custom/images/thumb/sphx_glr_polynomial_regression_thumb.png
    :alt:

  :ref:`sphx_glr_examples_Custom_polynomial_regression.py`

.. raw:: html

      <div class="sphx-glr-thumbnail-title">Polynomial regression</div>
    </div>


.. thumbnail-parent-div-close

.. raw:: html

    </div>


.. toctree::
   :hidden:

   /examples/Custom/neural_net
   /examples/Custom/polynomial_regression


.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-gallery

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download all examples in Python source code: Custom_python.zip </examples/Custom/Custom_python.zip>`

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download all examples in Jupyter notebooks: Custom_jupyter.zip </examples/Custom/Custom_jupyter.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_