airbnb_polyreg_data.py

# prepare train data
df_train_summary = df_train_2.groupby(['date']) \
                    .agg({'price': np.mean}).reset_index()
min_date = df_train_summary.date.min()
df_train_summary.date = df_train_summary.date - min_date
df_train_summary.date = df_train_summary.date.dt.days
df_train_summary.sample(frac=.01)

# prepare test data
df_test_2 = convert_date(df_test)
df_test_summary = df_test_2.groupby(['date']) \
                    .agg({'price': np.mean}).reset_index()
df_test_summary.date = df_test_summary.date - min_date
df_test_summary.date = df_test_summary.date.dt.days