mlflow attempt task 2 prediction

This commit is contained in:
Kamila 2022-05-15 12:45:13 +02:00
parent 941ac732d6
commit 6e04387e96

View File

@ -18,6 +18,7 @@ RUN pip3 install sklearn
RUN pip3 install pymongo RUN pip3 install pymongo
RUN pip3 install sacred RUN pip3 install sacred
RUN pip3 install mlflow RUN pip3 install mlflow
RUN pip3 install torch
CMD python3 data_expl.py CMD python3 data_expl.py
CMD python3 nn_train.py CMD python3 nn_train.py