.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "examples/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_examples_train_6kw_pmsyrm_curr_map_fem.py>`
        to download the full example code.

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

.. _sphx_glr_examples_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-22

.. code-block:: Python


    from pathlib import Path

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

    from motulator_gradnet import (
        Softmax,
        get_training_data,
        print_current_map_errors_fem,
        train_gradnet,
    )


.. GENERATED FROM PYTHON SOURCE LINES 23-24

Compute base values based on the nominal values.

.. GENERATED FROM PYTHON SOURCE LINES 24-28

.. 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 29-30

Set up the paths and parameters.

.. GENERATED FROM PYTHON SOURCE LINES 30-39

.. 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.npz"
    subsample = 10
    k = 6
    activation = Softmax


.. GENERATED FROM PYTHON SOURCE LINES 40-43

Train the model. The FEM data is on a uniform grid, so the loss is dominated by the
saturated region. The loss of the data points whose current magnitude is below 12 A
is weighted by 10 to improve the accuracy at low currents.

.. GENERATED FROM PYTHON SOURCE LINES 43-57

.. code-block:: Python


    if not trained_path.exists():
        train_gradnet(
            dataset_path=dataset_path,
            base=base,
            save_model_path=trained_path,
            k=k,
            embed_dim=48,
            epochs=3000,
            subsample=subsample,
            activation=activation,
            i_low=12,
            weight_low=10,
        )

.. GENERATED FROM PYTHON SOURCE LINES 58-59

Load the dataset for visualization comparison.

.. GENERATED FROM PYTHON SOURCE LINES 59-65

.. code-block:: Python


    # Get the training and validation data (complement) from the helper function
    _, val_data = get_training_data(
        str(dataset_path), base=base, subsample=subsample, other_keys=["theta_m", "tau_m"]
    )


.. GENERATED FROM PYTHON SOURCE LINES 66-67

Load the GradNet model and create its callable.

.. GENERATED FROM PYTHON SOURCE LINES 67-71

.. code-block:: Python


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


.. GENERATED FROM PYTHON SOURCE LINES 72-73

Print statistical error metrics on validation data.

.. GENERATED FROM PYTHON SOURCE LINES 73-76

.. code-block:: Python


    val_dict = dict(zip(["psi_s_dq", "i_s_dq", "theta_m", "tau_m"], val_data, strict=True))
    print_current_map_errors_fem(map_fcn=harm_map, raw_data=val_dict, base=base)


.. _sphx_glr_download_examples_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>`_