Skip to content

Instantly share code, notes, and snippets.

@erykml

erykml/ts_feature_engineering_4.py

Created July 28, 2022 20:26
Show Gist options
  • Save erykml/da648ec9442b8ef441a7e530b3bc83ff to your computer and use it in GitHub Desktop.
Save erykml/da648ec9442b8ef441a7e530b3bc83ff to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ts_feature_engineering_4.py
model_1 = LinearRegression().fit(X_1.iloc[:TRAIN_END],
y.iloc[:TRAIN_END])
results_df["model_1"] = model_1.predict(X_1)
results_df[["actuals", "model_1"]].plot(figsize=(16,4),
title="Fit using month dummies")
plt.axvline(date(2020, 1, 1), c="m", linestyle="--");
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment