Skip to content

Commit 820720b

Browse files
committed
[change] Remove main from src/multi_svr.py
1 parent c40d1fe commit 820720b

File tree

1 file changed

+0
-33
lines changed

1 file changed

+0
-33
lines changed

src/multi_svr.py

Lines changed: 0 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -41,36 +41,3 @@ def predict(self, X):
4141

4242
pred = np.column_stack(tuple(preds))
4343
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-
[0, 0],
61-
[0, 10],
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

Comments
 (0)