Commit 48c02996 authored by engel's avatar engel
Browse files

Add otimization to classifier for SVMand MLP

parent a6980e66
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -86,9 +86,9 @@ rnd_forr = RandomForestClassifier(criterion='entropy', max_depth= 9, min_samples
naive_bayes = CategoricalNB() # no smoothing 
naive_bayes_ws = CategoricalNB(alpha=1.0) # smoothing with alpha = 1.0
# SVM Kernel
svc = SVC(kernel="rbf", gamma=5, decision_function_shape='ovo', C=1)
svc = SVC(kernel="rbf", gamma=1, decision_function_shape='ovo', C=1)
# Neuronal Network
mlp =  MLPClassifier(alpha=1, max_iter=1000)
mlp =  MLPClassifier(activation="tanh" ,alpha=0.5380556432206247,hidden_layer_sizes=(473, 441), learning_rate="constant", solver="sgd", max_iter=1000)

# choose classifier
classifiers = [