INNER CODE UNIT · Python
get_train_test
alteryx/open_source_demos · predict-daily-temperature/utils.py:71
def get_train_test(df):
split_point = int(df.shape[0]*.7)
training_data = df[:split_point]
test_data = df[split_point:]
return training_data, test_data
def add_delayed_feature(df, col_to_delay, delay_length):
target_delay_training = df[col_to_delay].shift(delay_length)
target_delay_training.name = 'target_delay'
return pd.concat([df, target_delay_training], axis=1)
def train_and_fit_random_forest_regressor(X_train, y_train, X_test, y_test):
reg = RandomForestRegressor(n_estimators=100)
reg.fit(X_train, y_train)