Commit 8a5e66c1 authored by Marco Chierici's avatar Marco Chierici
Browse files

Added k='all' in SelectKBest (Nicole didn't commit!)

parent 0492a38e
......@@ -167,7 +167,7 @@ for n in range(CV_N):
elif RANK_METHOD == 'randomForest' :
ranking_tmp = np.argsort(forest.feature_importances_)[::-1]
elif RANK_METHOD == 'KBest':
selector = SelectKBest(f_classif)
selector = SelectKBest(f_classif, k='all')
selector.fit(x_tr, y_tr)
ranking_tmp = np.argsort( -np.log10(selector.pvalues_) )[::-1]
elif RANK_METHOD == 'rankList':
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment