def train(self):
clf = SVR(C=1.0, epsilon=0.1, cache_size=1000)
X, y, = self._get_data('training-2016-12-01-2017-02-28.csv')
# Fit the model
clf.fit(X, y)
# Pickle the model so we can save and reuse it
s = pickle.dumps(clf)
# Save the model to a file
f = open('finish_pos.model', 'wb')
f.write(s)
f.close()
评论列表
文章目录