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

changed hyperparameters

This commit is contained in:
Anne Lorenz 2018-09-27 09:21:02 +02:00
parent 549f21657c
commit 03d96987b8
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
NaiveBayes_simple.py
View File

@ -3,6 +3,8 @@ Naive Bayes Classifier
======================
basic implementation of naive bayes.
prints out probabilities for classes.
needed for interactive labeling.
'''
from CsvHandler import CsvHandler

6
SVM.py
View File

@ -52,10 +52,10 @@ class SVM:
pipeline = Pipeline([('perc', selector), ('SVC', SVC())])
grid = GridSearchCV(pipeline, {'perc__percentile': [25, 50, 75],
grid = GridSearchCV(pipeline, {'perc__percentile': [50, 75],
'SVC__kernel': ['linear'],
'SVC__gamma': [0.000001, 0.00001, 0.0001, 0.001],
'SVC__C': [0.001, 0.01, 0.1, 1, 10]},
'SVC__gamma': [0.00001, 0.0001],
'SVC__C': [0.1, 1]},
cv=skf,
scoring=make_scorer(f1_score))