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 sacred
RUN pip3 install mlflow
RUN pip3 install torch
CMD python3 data_expl.py
CMD python3 nn_train.py