()
| 55 | |
| 56 | |
| 57 | def test_svm_classification(): |
| 58 | y_signed_train = (y_train * 2) - 1 |
| 59 | y_signed_test = (y_test * 2) - 1 |
| 60 | |
| 61 | for kernel in [RBF(gamma=0.05), Linear()]: |
| 62 | model = SVM(max_iter=500, kernel=kernel) |
| 63 | model.fit(X_train, y_signed_train) |
| 64 | predictions = model.predict(X_test) |
| 65 | assert accuracy(y_signed_test, predictions) >= 0.8 |
| 66 | |
| 67 | |
| 68 | def test_mlp(): |