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

    Click :ref:`here <sphx_glr_download_auto_examples_Time_Features_examples_plot_time_feature_importance.py>` to download the full example code
.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_Time_Features_examples_plot_time_feature_importance.py:


===================================================
      Feature Importance using Forest of Trees
===================================================

This example shows how to compute feature importance for a classification task using forest of trees




.. code-block:: python


    # import matplotlib
    # matplotlib.use('TkAgg')
    from os.path import isdir, join
    import sab_dataset
    import seaborn as sns
    sns.set()
    sns.set_context('paper')







Load the data : sab dataset



.. code-block:: python

    sab_dataset_dirpath = join('pySAB', 'sample_data') if isdir('pySAB') else join('..', '..', 'pySAB', 'sample_data')
    sab_dataset_filename = 'sab_dataset_rec_subject_id_040119_1153.p'
    rec_dataset = sab_dataset.load_sab_dataset(join(sab_dataset_dirpath, sab_dataset_filename))







Downsample the data



.. code-block:: python

    rec_dataset.downsample(2)





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

 Out:

 .. code-block:: none

    New sampling rate is 256.0


Construct the features from the SabDataset object - Select only 'hits' and 'correct rejects' trials and keep only
1 electrode of interest :



.. code-block:: python

    time_features = rec_dataset.create_features(trial_sel=(rec_dataset.hits | rec_dataset.correct_rejects))
    print(time_features)





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

 Out:

 .. code-block:: none

    Time Features subject_id_rec - 7 features, 180 time points, 356 trials
    2 labels : {1: 'Hits', 2: 'Correct rejects'}
    Feature types : Amp


Compute feature importance using forest of decision trees



.. code-block:: python

    time_features.compute_feature_importance([1, 2])




.. image:: /auto_examples/Time_Features_examples/images/sphx_glr_plot_time_feature_importance_001.png
    :class: sphx-glr-single-img




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


.. _sphx_glr_download_auto_examples_Time_Features_examples_plot_time_feature_importance.py:


.. only :: html

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



  .. container:: sphx-glr-download

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



  .. container:: sphx-glr-download

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


.. only:: html

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

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