Add progress bar

Bougeant · Oct 18, 2024 · 6520227 · 6520227
1 parent dc4a222
commit 6520227
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ml_inspector/partial_dependence.py b/ml_inspector/partial_dependence.py
@@ -4,6 +4,7 @@
 import pandas as pd
 from plotly import graph_objs as go
 from plotly.colors import DEFAULT_PLOTLY_COLORS
+from tqdm.auto import tqdm
 
 
 def select_feature_values(X, feature, max_nb_points):
@@ -136,7 +137,7 @@ def plot_partial_dependence(
     """
     plot_data = []
     visible = True
-    for feature in X.columns:
+    for feature in tqdm(X.columns, desc="Calculating partial dependence"):
         feature_values, all_predictions, all_impacts = partial_dependence(
             estimator,
             X,

diff --git a/pyproject.toml b/pyproject.toml
@@ -14,6 +14,7 @@ dependencies = [
     "pandas",
     "scikit-learn",
     "plotly",
+    "tqdm",
 ]
-Original file line number
+Diff line change
@@ Expand Up / @@ -14,6 +14,7 @@ dependencies = [ @@
         "pandas",
         "scikit-learn",
         "plotly",
+        "tqdm",
     ]
@@ Expand Down @@