Skip to content

Commit

Permalink
statsforecast.AutoCES model
Browse files Browse the repository at this point in the history
  • Loading branch information
attila-balint-kul committed Nov 30, 2023
1 parent e8cffeb commit e86b13f
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 0 deletions.
18 changes: 18 additions & 0 deletions models/sf-auto-ces/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
FROM python:3.11.6-slim-bookworm

WORKDIR /usr/local/app

COPY ./requirements.txt /usr/local/app/requirements.txt
RUN pip install --no-cache-dir --upgrade pip
RUN pip install --no-cache-dir -r /usr/local/app/requirements.txt

# Polars requires AVX2 CPU instructions, which are not available on the server
RUN pip uninstall -y polars
RUN pip install --no-cache-dir polars-lts-cpu

COPY ./src /usr/local/app/src

ENV ENFOBENCH_MODEL_SEASONALITY="1D"

EXPOSE 3000
CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "3000"]
2 changes: 2 additions & 0 deletions models/sf-auto-ces/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
enfobench>=0.3.5,<0.4.0
statsforecast==1.5.0
1 change: 1 addition & 0 deletions models/sf-naive-seasonal/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,5 +54,6 @@ def forecast(

# Instantiate your model
model = SeasonalNaiveModel(seasonality)

# Create a forecast server by passing in your model
app = server_factory(model)

0 comments on commit e86b13f

Please sign in to comment.