julian.kunkel
/
thesis-anne
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

changed hyperparameters in SVM

This commit is contained in:
Anne Lorenz 2018-09-20 09:40:17 +02:00
parent 5d4b65cbf7
commit 68ba9e83cb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
SVM.py
View File

@ -54,8 +54,8 @@ class SVM:
grid = GridSearchCV(pipeline, {'perc__percentile': [25, 50, 75, 100],
'SVC__kernel': ['linear','poly'],
'SVC__gamma': [0.001, 0.01],
'SVC__C': [0.1, 1]},
'SVC__gamma': [0.0001, 0.001, 0.01, 0.1],
'SVC__C': [0.0001, 0.001, 0.1]},
cv=skf,
scoring=make_scorer(f1_score))