diff options
Diffstat (limited to 'KM-SVM/KM-SVM.py')
| -rw-r--r-- | KM-SVM/KM-SVM.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/KM-SVM/KM-SVM.py b/KM-SVM/KM-SVM.py index 71bf984..70cad7f 100644 --- a/KM-SVM/KM-SVM.py +++ b/KM-SVM/KM-SVM.py @@ -117,7 +117,7 @@ def SVM(train_X, train_Y, test_X, test_Y): train_Y = [np.where(r == 1)[0][0] for r in train_Y] test_Y = [np.where(r == 1)[0][0] for r in test_Y] t1 = time.time() - clf = svm.SVC(decision_function_shape='ovr', max_iter=900, kernel='rbf') + clf = svm.SVC(decision_function_shape='ovr', max_iter=1000, kernel='rbf') model = clf.fit(train_X, train_Y) y_hat = model.predict(test_X) acc = accuracy_score(test_Y, y_hat) |