Skip to content
Snippets Groups Projects
Commit a8ffedcb authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Debigging

parent 540ea254
Branches
No related tags found
No related merge requests found
Pipeline #11596 failed
...@@ -25,12 +25,13 @@ def get_feature_importances(result, feature_ids=None, view_names=None,): ...@@ -25,12 +25,13 @@ def get_feature_importances(result, feature_ids=None, view_names=None,):
""" """
feature_importances = {} feature_importances = {}
for classifier_result in result: for classifier_result in result:
print((classifier_result.classifier_name)) print(classifier_result.classifier_name)
if isinstance(classifier_result, MonoviewResult): if isinstance(classifier_result, MonoviewResult):
if classifier_result.view_name not in feature_importances: if classifier_result.view_name not in feature_importances:
feature_importances[classifier_result.view_name] = pd.DataFrame( feature_importances[classifier_result.view_name] = pd.DataFrame(
index=feature_ids[classifier_result.view_index]) index=feature_ids[classifier_result.view_index])
if hasattr(classifier_result.clf, 'feature_importances_'): if hasattr(classifier_result.clf, 'feature_importances_'):
print(classifier_result.classifier_name)
feature_importances[classifier_result.view_name][ feature_importances[classifier_result.view_name][
classifier_result.classifier_name] = classifier_result.clf.feature_importances_ classifier_result.classifier_name] = classifier_result.clf.feature_importances_
else: else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment