Beispiel #1
0
def fit(dataframe, target, city, station):
    ''' Train the random forest to predict bikes or spaces. '''
    X, Y = munging.split(dataframe, target)
    regressor = RandomForestRegressor(n_estimators=12, n_jobs=-1, max_depth=10)
    regressor.fit(X, Y)
    save_regressor(regressor, 'forest', target, city, station)
Beispiel #2
0
def fit(dataframe, target, city, station):
    ''' Train the decision tree to predict bikes or spaces. '''
    X, Y = munging.split(dataframe, target)
    regressor = DecisionTreeRegressor()
    regressor.fit(X, Y)
    save_regressor(regressor, 'tree', target, city, station)