Beispiel #1
0
    def setUp(self):
        self.df = select_mungedata(2, 1, '2013-1-1', '2015-1-1')
        self.xtest = select_features('2015-1-1', '2015-1-10')
        self.actuals = select_mungedata_simple(2, 1, '2015-1-1', '2015-1-10')
        self.daily_avg = self.df.waittime.resample('D', how='mean')

        model = RandomForestRegressor(n_jobs=-1,
                                      n_estimators=2,
                                      random_state=1)
        self.im = IncrementalModel(self.df, model, categoricals=['event'])
        self.im.predict(self.xtest)