# feature selection
f_selector = SelectKBest(score_func=mutual_info_regression, k='all')