Coverage for biobb_pytorch / mdae / explainability / sensitivity_analysis.py: 100%

1from sklearn.feature_selection import mutual_info_regression 

2import pandas as pd 

3 

4 

5def make_mi_scores(X, y, discrete_features): 

6 mi_scores = mutual_info_regression(X, y, discrete_features=discrete_features) 

7 mi_scores = pd.Series(mi_scores, name="MI Scores", index=X.columns) 

8 mi_scores = mi_scores.sort_values(ascending=False) 

9 return mi_scores 

10 

11# mi_scores = make_mi_scores(X_pca, y, discrete_features=False) 

12# mi_scores