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

Feature importance correction

parent 61d553a5
No related branches found
No related tags found
No related merge requests found
Pipeline #11641 failed
......@@ -6,6 +6,10 @@ import os
def execute(config_path=None): # pragma: no cover
import sys
if not sys.warnoptions:
import warnings
warnings.simplefilter("ignore")
from summit.multiview_platform import exec_classif
if config_path is None:
sum = exec_classif.Summit(config_path=sys.argv[1:])
......
......@@ -92,8 +92,9 @@ def publish_feature_importances(feature_importances, directory, database_name,
feature_importances_df = pd.DataFrame(index=indices, columns=columns)
for df in importance_dfs:
feature_importances_df = feature_importances_df.combine_first(df)
feature_importances_df = feature_importances_df/feature_importances_df.sum(axis=0)
for col in feature_importances_df.columns:
if feature_importances_df[col].sum()!=0:
feature_importances_df[col] /= feature_importances_df[col].sum()
feature_std_df = pd.DataFrame(index=indices, columns=columns)
if len(std_dfs)>0:
for df in std_dfs:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment