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

added random state to split in function

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-05 14:15:43 +01:00
parent e5f882841e
commit 53c229f480
3 changed files with 24 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/ml/ml_functions/MultClfs.py
View file

@ -215,7 +215,7 @@ def MultModelsCl(input_df, target, skf_cv
, ('Naive Bayes' , BernoulliNB() )
, ('Passive Aggresive' , PassiveAggressiveClassifier(**rs, **njobs) )
, ('QDA' , QuadraticDiscriminantAnalysis() )
, ('Random Forest' , RandomForestClassifier(**rs, n_estimators = 1000 ) )
, ('Random Forest' , RandomForestClassifier(**rs, n_estimators = 1000, **njobs ) )
, ('Random Forest2' , RandomForestClassifier(min_samples_leaf = 5
, n_estimators = 1000
, bootstrap = True