.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "drive_examples/gradnet/train_6kw_pmsyrm_curr_map_fem.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_drive_examples_gradnet_train_6kw_pmsyrm_curr_map_fem.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_drive_examples_gradnet_train_6kw_pmsyrm_curr_map_fem.py:


5.6-kW PM-SyRM, train current map, FEM data with spatial harmonics
==================================================================

This example trains a GradNet current map for a four-pole 5.6-kW PM synchronous
reluctance machine (Baldor ECS101M0H7EF4) from a FEM dataset with spatial harmonics.

.. GENERATED FROM PYTHON SOURCE LINES 9-15

.. code-block:: Python


    from pathlib import Path

    import motulator.drive.gradnet as gn
    from motulator.drive import utils


.. GENERATED FROM PYTHON SOURCE LINES 16-17

Set nominal and base values.

.. GENERATED FROM PYTHON SOURCE LINES 17-21

.. code-block:: Python


    nom = utils.NominalValues(U=460, I=8.8, f=60, P=5.6e3, tau=29.7)
    base = utils.BaseValues.from_nominal(nom, n_p=2)


.. GENERATED FROM PYTHON SOURCE LINES 22-23

Set up the paths and parameters.

.. GENERATED FROM PYTHON SOURCE LINES 23-32

.. code-block:: Python


    p = Path(__file__).resolve().parent if "__file__" in globals() else Path.cwd()

    dataset_path = p / "datasets/baldor_fem.npz"
    trained_path = p / "trained_models" / "baldor_fem_curr_map_harm_softmax_d48_sub10.pth"
    subsample = 10
    k = 6
    activation = gn.Softmax


.. GENERATED FROM PYTHON SOURCE LINES 33-34

Train the model.

.. GENERATED FROM PYTHON SOURCE LINES 34-46

.. code-block:: Python


    if not trained_path.exists():
        gn.train_gradnet(
            dataset_path=dataset_path,
            base=base,
            save_model_path=trained_path,
            k=k,
            embed_dim=48,
            epochs=1000,
            subsample=subsample,
            activation=activation,
        )

.. GENERATED FROM PYTHON SOURCE LINES 47-48

Load the dataset for visualization comparison.

.. GENERATED FROM PYTHON SOURCE LINES 48-62

.. code-block:: Python


    # Get the training and validation data (complement) from the helper function
    # Note: get_training_data returns (psi, i, ...), but we need (i, psi, ...)
    (trn_psi, trn_i, trn_theta, trn_tau), (val_psi, val_i, val_theta, val_tau) = (
        gn.get_training_data(
            str(dataset_path),
            base=base,
            subsample=subsample,
            other_keys=["theta_m", "tau_m"],
        )
    )
    trn_data = (trn_i, trn_psi, trn_theta, trn_tau)
    val_data = (val_i, val_psi, val_theta, val_tau)


.. GENERATED FROM PYTHON SOURCE LINES 63-64

Load the GradNet model and create its callable.

.. GENERATED FROM PYTHON SOURCE LINES 64-68

.. code-block:: Python


    model = gn.load_gradnet(trained_path, activation=activation)
    harm_map = gn.CurrentMapWithHarmonics(model, k=k)


.. GENERATED FROM PYTHON SOURCE LINES 69-70

Print statistical error metrics on validation data.

.. GENERATED FROM PYTHON SOURCE LINES 70-78

.. code-block:: Python


    val_dict = {
        "i_s_dq": val_i,
        "psi_s_dq": val_psi,
        "theta_m": val_theta,
        "tau_m": val_tau,
    }
    gn.print_current_map_errors_fem(map_fcn=harm_map, raw_data=val_dict, base=base)


.. _sphx_glr_download_drive_examples_gradnet_train_6kw_pmsyrm_curr_map_fem.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

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

      :download:`Download Jupyter notebook: train_6kw_pmsyrm_curr_map_fem.ipynb <train_6kw_pmsyrm_curr_map_fem.ipynb>`

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

      :download:`Download Python source code: train_6kw_pmsyrm_curr_map_fem.py <train_6kw_pmsyrm_curr_map_fem.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: train_6kw_pmsyrm_curr_map_fem.zip <train_6kw_pmsyrm_curr_map_fem.zip>`


.. only:: html

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

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