зеркало из https://github.com/mozilla/bugbug.git
Use human readable feature names when classifying too
This commit is contained in:
Родитель
e7ed31940c
Коммит
481917afac
|
@ -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(
|
||||
|
|
Загрузка…
Ссылка в новой задаче