def setUp(self):
self.predictor = TrendPredictor()
self.X_train, self.X_test, self.y_train, self.y_test = \
Business.train_test_set(DataReader().sample_businesses())
self.predictor.fit(self.X_train, self.y_train)
def svr_trend_prediction():
logger = logging.getLogger('trend-prediction')
logger.setLevel(logging.INFO)
logger.addHandler(logging.FileHandler('../logs/trend_prediction.log'))
predictor = TrendPredictor()
X_train, X_test, y_train, y_test = \
Business.train_test_set(DataReader().businesses())
predictor.fit(X_train, y_train)
predictor.save()
logger.info(predictor)
logger.info('mean squared error: %s ' %
predictor.mean_squared_error(X_test, y_test))
logger.info('r2 score: %s ' % predictor.r2_score(X_test, y_test))
