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))