21 lines
331 B
Docker
21 lines
331 B
Docker
FROM python:3.8-slim-buster
|
|
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
figlet \
|
|
unzip \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
RUN pip install --upgrade pip
|
|
RUN pip install torch pandas numpy scikit-learn kaggle
|
|
|
|
COPY . /app
|
|
COPY ./train.py ./
|
|
COPY ./prediction.py ./
|
|
COPY ./Sales.csv ./
|
|
|
|
CMD python train.py
|
|
|
|
|
|
|