diff --git a/summit/multiview_platform/monoview_classifiers/adaboost.py b/summit/multiview_platform/monoview_classifiers/adaboost.py index cd8ce3db0b769e7ad99032487d94da010988138b..412e9a19fe2c5b523bb358018dc8381acf488dd8 100644 --- a/summit/multiview_platform/monoview_classifiers/adaboost.py +++ b/summit/multiview_platform/monoview_classifiers/adaboost.py @@ -64,11 +64,12 @@ class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier): [step_pred for step_pred in self.staged_predict(X)]) return pred - def get_interpretation(self, directory, base_file_name, y_test, + def get_interpretation(self, directory, base_file_name, y_test, feature_ids, multi_class=False): # pragma: no cover interpretString = "" interpretString += self.get_feature_importance(directory, - base_file_name) + base_file_name, + feature_ids) interpretString += "\n\n Estimator error | Estimator weight\n" interpretString += "\n".join( [str(error) + " | " + str(weight / sum(self.estimator_weights_)) for diff --git a/summit/multiview_platform/monoview_classifiers/gradient_boosting.py b/summit/multiview_platform/monoview_classifiers/gradient_boosting.py index e242dee80c6c1ef76daacd2c43d4c178b8f4c495..77242502ca370fabba2f51df167774a2c3ac24e2 100644 --- a/summit/multiview_platform/monoview_classifiers/gradient_boosting.py +++ b/summit/multiview_platform/monoview_classifiers/gradient_boosting.py @@ -76,14 +76,15 @@ class GradientBoosting(GradientBoostingClassifier, BaseMonoviewClassifier): [step_pred for step_pred in self.staged_predict(X)]) return pred - def get_interpretation(self, directory, base_file_name, y_test, + def get_interpretation(self, directory, base_file_name, y_test, feature_ids, multi_class=False): interpretString = "" if multi_class: return interpretString else: interpretString += self.get_feature_importance(directory, - base_file_name) + base_file_name, + feature_ids) step_test_metrics = np.array( [self.plotted_metric.score(y_test, step_pred) for step_pred in self.step_predictions]) diff --git a/summit/multiview_platform/monoview_classifiers/random_forest.py b/summit/multiview_platform/monoview_classifiers/random_forest.py index c0ebaaa570e33e6d0fa2a92944a16b7f7ccecb99..f0d3578c3c81ac5f11c6baefd6b845d984986314 100644 --- a/summit/multiview_platform/monoview_classifiers/random_forest.py +++ b/summit/multiview_platform/monoview_classifiers/random_forest.py @@ -34,10 +34,11 @@ class RandomForest(RandomForestClassifier, BaseMonoviewClassifier): ["gini", "entropy"], [random_state]] self.weird_strings = {} - def get_interpretation(self, directory, base_file_name, y_test, + def get_interpretation(self, directory, base_file_name, y_test, feature_ids, multiclass=False): interpret_string = "" interpret_string += self.get_feature_importance(directory, - base_file_name) + base_file_name, + feature_ids) return interpret_string