restructure notebook and add autoarima + prophet

This commit is contained in:
Tristan de Waard 2025-06-12 16:16:26 +02:00
parent c5746ea7b1
commit 127ff6b237
6 changed files with 3169 additions and 1182 deletions

1
.gitignore vendored
View File

@ -1,3 +1,4 @@
.venv/ .venv/
build/ build/
*.egg-info/ *.egg-info/
__pycache__/

View File

@ -43,4 +43,4 @@ Look in `tsforecast.ipynb`
- SARIMA - SARIMA
- XGBoost - XGBoost
- LSTM - 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)

Binary file not shown.

Before

Width:  |  Height:  |  Size: 504 KiB

After

Width:  |  Height:  |  Size: 679 KiB

View File

@ -8,8 +8,10 @@ authors = [
] ]
requires-python = ">=3.12,<3.13" requires-python = ">=3.12,<3.13"
dependencies = [ dependencies = [
"numpy", "numpy==1.26.4",
"pandas", "pandas",
"pmdarima",
"prophet",
"seaborn", "seaborn",
"xgboost", "xgboost",
"matplotlib", "matplotlib",
@ -28,3 +30,6 @@ dependencies = [
[build-system] [build-system]
requires = ["setuptools", "wheel"] requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"
[tool.poetry.dependencies]
numpy = "1.26.4"

File diff suppressed because one or more lines are too long

1602
uv.lock Normal file

File diff suppressed because it is too large Load Diff