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

added all classification algorithms params for gridsearch

Browse source
This commit is contained in:
Tanushree Tunstall 2022-03-21 13:51:20 +00:00
parent d012542435
commit 0c4f1e1e5f
8 changed files with 503 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

5
imports.py
View file

@ -81,6 +81,11 @@ njobs = {'n_jobs': 10}
skf_cv = StratifiedKFold(n_splits = 10
#, shuffle = False, random_state= None)
, shuffle = True,**rs)
rskf_cv = RepeatedStratifiedKFold(n_splits = 10
, n_repeats=3
#, shuffle = False, random_state= None)
#, shuffle = True
,**rs)
#my_mcc = make_scorer({'mcc':make_scorer(matthews_corrcoef})
mcc_score_fn = {'mcc': make_scorer(matthews_corrcoef)}