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

saving work

Browse source
This commit is contained in:
Tanushree Tunstall 2022-03-18 17:50:24 +00:00
parent ab1508e9fb
commit 824c2f041c
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
hyperparams.py
View file

@ -91,10 +91,6 @@ gs_gb = GridSearchCV(estimator=XGBClassifier(n_jobs=-1),
n_jobs=4,
refit='recall')
#%%
#%%
gs_mlp = MLPClassifier(max_iter=600)
clf = GridSearchCV(gs_mlp, parameter_space, n_jobs= 10, cv = 10)