tanu/ML_AI_training
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

saved the UQ_FS_eg with almost complete output, still some minor fixes to do

Browse source
This commit is contained in:
Tanushree Tunstall 2022-05-23 03:35:35 +01:00
parent 1436557287
commit a420822a93
3 changed files with 173 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

7
UQ_pnca_ML.py
View file

@ -84,6 +84,7 @@ from imblearn.under_sampling import EditedNearestNeighbours
from sklearn.model_selection import GridSearchCV
from sklearn.base import BaseEstimator
import json
scoring_fn = ({'accuracy' : make_scorer(accuracy_score)
, 'fscore' : make_scorer(f1_score)
@ -101,10 +102,8 @@ skf_cv = StratifiedKFold(n_splits = 10
, shuffle = True,**rs)
rskf_cv = RepeatedStratifiedKFold(n_splits = 10
, n_repeats=3
#, shuffle = False, random_state= None)
#, shuffle = True
,**rs)
, n_repeats = 3
, **rs)
mcc_score_fn = {'mcc': make_scorer(matthews_corrcoef)}
jacc_score_fn = {'jcc': make_scorer(jaccard_score)}