restructure notebook and add autoarima + prophet

2025-06-12 16:16:26 +02:00 · 2025-06-12 16:16:26 +02:00 · 127ff6b237
commit 127ff6b237
parent c5746ea7b1
6 changed files with 3169 additions and 1182 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,3 +1,4 @@
 .venv/
 build/
 *.egg-info/
+__pycache__/
--- a/README.md
+++ b/README.md
@ -43,4 +43,4 @@ Look in `tsforecast.ipynb`
  - SARIMA
  - XGBoost
  - LSTM
- Try to beat my MAE on test set of ~0.31 (I used LSTM)
+- Try to beat my MAE on test set of ~0.30 (I used LSTM)
--- a/price_actual_over_time.png
+++ b/price_actual_over_time.png
--- a/pyproject.toml
+++ b/pyproject.toml
@ -8,8 +8,10 @@ authors = [
 ]
 requires-python = ">=3.12,<3.13"
 dependencies = [
-    "numpy",
+    "numpy==1.26.4",
    "pandas",
+    "pmdarima",
+    "prophet",
    "seaborn",
    "xgboost",
    "matplotlib",
@ -28,3 +30,6 @@ dependencies = [
 [build-system]
 requires = ["setuptools", "wheel"]
 build-backend = "setuptools.build_meta"
+
+[tool.poetry.dependencies]
+numpy = "1.26.4"
--- a/tsforecast.ipynb
+++ b/tsforecast.ipynb
--- a/uv.lock
+++ b/uv.lock