We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c40d1fe commit 820720bCopy full SHA for 820720b
src/multi_svr.py
@@ -41,36 +41,3 @@ def predict(self, X):
41
42
pred = np.column_stack(tuple(preds))
43
return pred
44
-
45
46
47
48
-if __name__ == '__main__':
49
- from sklearn import metrics
50
- X = [
51
- [0, 0],
52
- [0, 10],
53
- [1, 10],
54
- [1, 20],
55
- [1, 30],
56
- [1, 40]
57
- ]
58
59
- y = [
60
61
62
- [2, 10],
63
- [2, 20],
64
- [2, 30],
65
- [2, 40]
66
67
68
- regressor = MutilSVR(kernel='linear')
69
70
- regressor.fit(X, y)
71
72
- pred_y = regressor.predict(X)
73
- errs = metrics.mean_squared_error(y, pred_y, multioutput='raw_values')
74
75
- print('pred_y:', pred_y)
76
- print('errs:', errs)
0 commit comments