Skip to content

Callbacks

Callbacks for feature selection algorithms.

plot_progress_callback(selector, *args, **kwargs)

Plot the feature selection progress during the algorithm execution.

Parameters:

  • selector (object) –

    The feature selector object.

Source code in felimination/callbacks.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
def plot_progress_callback(selector, *args, **kwargs):
    """Plot the feature selection progress during the algorithm execution.

    Parameters
    ----------
    selector : object
        The feature selector object.
    """
    from IPython import display
    from matplotlib import pyplot as plt

    display.clear_output(wait=True)
    selector.plot()
    plt.show()