Update xgb.get_importances for non-string feature names

mxbi · Dec 24, 2019 · 74bc457 · 74bc457
1 parent b8a2dda
commit 74bc457
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlcrate/xgb.py b/mlcrate/xgb.py
@@ -12,6 +12,8 @@ def get_importances(model, features):
     importance -- A list of (feature, importance) tuples representing sorted importance
     """
 
+    features = [str(f) for f in features]
+
     for feature in features:
         assert '\n' not in feature and '\t' not in feature, "\\n and \\t cannot be in feature names"