Updated to task 5

This commit is contained in:
Grzegorz Gapiński 2023-05-12 23:33:56 +02:00
parent 391b74a662
commit f46f5dffec

View File

@ -8,6 +8,8 @@ RUN apt-get install -y unzip
RUN pip3 install pandas
RUN pip3 install kaggle
RUN pip3 install scikit-learn
RUN pip3 install numpy
RUN pip3 install tensorflow
RUN ln -s ~/.local/bin/kaggle /usr/bin/kaggle
ENV KAGGLE_USERNAME="grzegorzgapiski"
@ -19,4 +21,12 @@ COPY ./data_preparation.py ./
COPY ./2019-european-parliament-election-in-poland-data ./
CMD ./data_preparation.py
COPY model_train.py /app/model_train.py
COPY model_test.py /app/model_test.py
CMD ./data_preparation.py
CMD ["python3", "model_train.py"]
CMD ["python3", "model_test.py"]