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

.. only:: html

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

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

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

.. _sphx_glr_gallery_plot_prototype_selection.py:


Prototype selection
===================

Visualise the results of the prototype selection step.

We use the data from the CalMS21 Task1 dataset (Sun et al., 2021).

.. GENERATED FROM PYTHON SOURCE LINES 11-14

Import and configure modules
----------------------------
Import the necessary modules and configure the plotting settings.

.. GENERATED FROM PYTHON SOURCE LINES 14-25

.. code-block:: Python


    import matplotlib.pyplot as plt
    import numpy as np
    from huggingface_hub import hf_hub_download
    from scipy.spatial.distance import squareform

    import lisbet.plotting as betp

    # Configure
    plt.rcParams["figure.constrained_layout.use"] = True








.. GENERATED FROM PYTHON SOURCE LINES 26-30

Fetch the sample data from HuggingFace
--------------------------------------
Fetch and load the information file containing the results of the prototype selection.
This file is generated by the command ``betman prototype_selection``.

.. GENERATED FROM PYTHON SOURCE LINES 30-37

.. code-block:: Python

    data_path = hf_hub_download(
        repo_id="gchindemi/lisbet-examples",
        filename="prototype_selection/CalMS21_Task1/info_hmmbest_6_32.npz",
        repo_type="dataset",
    )
    hmm_info = np.load(data_path)








.. GENERATED FROM PYTHON SOURCE LINES 38-41

Visualise the data
------------------
Plot the silhouette profile and the prototype selection results.

.. GENERATED FROM PYTHON SOURCE LINES 41-88

.. code-block:: Python

    fig, axs = plt.subplots(
        nrows=2,
        ncols=2,
        width_ratios=[2, 16],
        height_ratios=[4, 16],
        figsize=(8, 6),
    )

    # Share axes
    axs[0, 0].sharey(axs[0, 1])
    axs[0, 1].sharex(axs[1, 1])

    # Customize layout
    fig.align_ylabels(axs[:, 0])

    betp.plot_slh_score(
        hmm_info["all_n_clusters"],
        hmm_info["all_score"],
        hmm_info["best_n_clusters"],
        hmm_info["best_score"],
        axs[0, 0],
    )

    betp.plot_slh_profile(
        distance=squareform(hmm_info["cond_dist_matrix"]),
        link_matrix=hmm_info["link_matrix"],
        cluster_labels=hmm_info["best_labels"],
        ax=axs[0, 1],
    )

    betp.plot_dendrogram(
        hmm_info["link_matrix"],
        cluster_labels=hmm_info["best_labels"],
        ax=axs[1, 0],
    )

    betp.plot_heatmap(
        squareform(hmm_info["cond_dist_matrix"]),
        hmm_info["link_matrix"],
        hmm_info["best_labels"],
        hmm_info["prototypes"],
        ax=axs[1, 1],
    )

    # Finalize plot
    axs[0, 0].legend(frameon=False)
    axs[1, 1].legend(frameon=False)



.. image-sg:: /gallery/images/sphx_glr_plot_prototype_selection_001.png
   :alt: Average, Profile
   :srcset: /gallery/images/sphx_glr_plot_prototype_selection_001.png
   :class: sphx-glr-single-img


.. rst-class:: sphx-glr-script-out

 .. code-block:: none


    <matplotlib.legend.Legend object at 0x700dfca36e70>




.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 1.936 seconds)


.. _sphx_glr_download_gallery_plot_prototype_selection.py:

.. only:: html

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

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

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

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

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

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

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


.. only:: html

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

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