add SklearnModel (#812)

Nixtla · Apr 15, 2024 · 791a2cb · 791a2cb
1 parent 19b6564
commit 791a2cb
Show file tree

Hide file tree

Showing 7 changed files with 1,048 additions and 51 deletions.
diff --git a/dev/environment.yml b/dev/environment.yml
@@ -13,12 +13,14 @@ dependencies:
   - pip
   - prophet
   - pyarrow
+  - scikit-learn
   - scipy>=1.7.3
   - statsmodels>=0.13.2
   - tabulate
   - threadpoolctl
   - plotly
   - pip:
+    - datasetsforecast
     - fugue[dask,ray]
     - nbdev
     - plotly-resampler

diff --git a/dev/local_environment.yml b/dev/local_environment.yml
@@ -12,12 +12,14 @@ dependencies:
   - pip
   - prophet
   - pyarrow
+  - scikit-learn
   - scipy>=1.7.3
   - statsmodels>=0.13.2
   - tabulate
   - threadpoolctl
   - plotly
   - pip:
+    - datasetsforecast
     - nbdev
     - plotly-resampler
     - polars