:py:mod:`rofunc.learning.ml.tpgmr`
==================================

.. py:module:: rofunc.learning.ml.tpgmr

.. autodoc2-docstring:: rofunc.learning.ml.tpgmr
   :allowtitles:

Module Contents
---------------

Classes
~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`TPGMR <rofunc.learning.ml.tpgmr.TPGMR>`
     - .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMR
          :summary:
   * - :py:obj:`TPGMRBi <rofunc.learning.ml.tpgmr.TPGMRBi>`
     - .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMRBi
          :summary:

API
~~~

.. py:class:: TPGMR(demos_x, task_params, nb_states: int = 4, reg: float = 0.001, plot=False)
   :canonical: rofunc.learning.ml.tpgmr.TPGMR

   Bases: :py:obj:`rofunc.learning.ml.tpgmm.TPGMM`

   .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMR

   .. rubric:: Initialization

   .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMR.__init__

   .. py:method:: gmm_learning()
      :canonical: rofunc.learning.ml.tpgmr.TPGMR.gmm_learning

      .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMR.gmm_learning

   .. py:method:: fit()
      :canonical: rofunc.learning.ml.tpgmr.TPGMR.fit

   .. py:method:: reproduce(model, show_demo_idx)
      :canonical: rofunc.learning.ml.tpgmr.TPGMR.reproduce

   .. py:method:: generate(model: pbdlib.HMM, ref_demo_idx: int, task_params: dict) -> numpy.ndarray
      :canonical: rofunc.learning.ml.tpgmr.TPGMR.generate

.. py:class:: TPGMRBi(demos_left_x, demos_right_x, task_params, plot=False)
   :canonical: rofunc.learning.ml.tpgmr.TPGMRBi

   Bases: :py:obj:`rofunc.learning.ml.tpgmr.TPGMR`

   .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMRBi

   .. rubric:: Initialization

   .. autodoc2-docstring:: rofunc.learning.ml.tpgmr.TPGMRBi.__init__

   .. py:method:: fit()
      :canonical: rofunc.learning.ml.tpgmr.TPGMRBi.fit

   .. py:method:: reproduce(models, show_demo_idx)
      :canonical: rofunc.learning.ml.tpgmr.TPGMRBi.reproduce

   .. py:method:: generate(models, ref_demo_idx: int, task_params: dict) -> typing.Tuple[numpy.ndarray, numpy.ndarray]
      :canonical: rofunc.learning.ml.tpgmr.TPGMRBi.generate