diff --git a/docs/reference/callbacks.md b/docs/reference/callbacks.md new file mode 100644 index 0000000..66a8dd8 --- /dev/null +++ b/docs/reference/callbacks.md @@ -0,0 +1 @@ +::: felimination.callbacks diff --git a/felimination/callbacks.py b/felimination/callbacks.py index cd772ee..7fb7ff7 100644 --- a/felimination/callbacks.py +++ b/felimination/callbacks.py @@ -1,5 +1,14 @@ +"""Callbacks for feature selection algorithms.""" + + def plot_progress_callback(selector, *args, **kwargs): - """Plot the feature selection progress.""" + """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