fix formatting

snehakumari321 · Mar 5, 2023 · 1173719 · 1173719
1 parent 8997080
commit 1173719
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/train_random_forest/run.py b/src/train_random_forest/run.py
@@ -161,8 +161,8 @@ def go(args):
 
 def plot_feature_importance(pipe, feat_names):
     # We collect the feature importance for all non-nlp features first
-    feat_imp = pipe["random_forest"].
-    feature_importances_[: len(feat_names) - 1]
+    feat_imp = pipe["random_forest"].feature_importances_[
+               : len(feat_names) - 1]
     # For the NLP feature we sum across all the TF-IDF dimensions into a global
     # NLP importance
     nlp_importance = sum(