()
| 113 | plot_margin(X_train[y_train==1], X_train[y_train==-1], clf) |
| 114 | |
| 115 | def test_non_linear(): |
| 116 | X1, y1, X2, y2 = gen_non_lin_separable_data() |
| 117 | X_train, y_train = split_train(X1, y1, X2, y2) |
| 118 | X_test, y_test = split_test(X1, y1, X2, y2) |
| 119 | |
| 120 | # X_train = np.load('inputClf/X_train.npy') |
| 121 | # y_train = np.load('inputClf/y_train.npy') |
| 122 | # X_test = np.load('inputClf/X_test.npy') |
| 123 | # y_test = np.load('inputClf/y_test.npy') |
| 124 | clf = SVM(gaussian_kernel, C=1) |
| 125 | clf.fit(X_train, y_train) |
| 126 | |
| 127 | y_predict = clf.predict(X_test) |
| 128 | correct = np.sum(y_predict == y_test) |
| 129 | print "%d out of %d predictions correct" % (correct, len(y_predict)) |
| 130 | |
| 131 | plot_contour(X_train[y_train==1], X_train[y_train==-1], clf) |
| 132 | |
| 133 | def test_soft(): |
| 134 | X1, y1, X2, y2 = gen_lin_separable_overlap_data() |
no test coverage detected