Use human readable feature names when classifying too

This commit is contained in:
Marco Castelluccio 2019-07-09 21:22:08 +02:00
Родитель e7ed31940c
Коммит 481917afac
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -337,7 +337,7 @@ class Model:
int(index) for importance, index, is_positive in top_importances
]
feature_names = self.get_feature_names()
feature_names = self.get_human_readable_feature_names()
with io.StringIO() as out:
p = shap.force_plot(

Просмотреть файл

@ -107,7 +107,7 @@ class CommitClassifier(object):
commits[-1], probabilities=True, importances=True
)
feature_names = self.model.get_feature_names()
feature_names = self.model.get_human_readable_feature_names()
features = []
for i, (val, feature_index, is_positive) in enumerate(