This commit is contained in:
Alicja Szulecka 2024-05-06 22:08:05 +02:00
parent ed9927d7a1
commit c4ce89938c
2 changed files with 18 additions and 3 deletions

15
mlflow/Dockerfile Normal file
View File

@ -0,0 +1,15 @@
FROM python:3.10
RUN pip install --upgrade pip
RUN pip3 install mlflow
RUN pip3 install scikit-learn
RUN pip3 install pandas
RUN pip3 install numpy
RUN pip3 install torch
COPY mlflow_model.py .
COPY mlflow_prediction.py .
COPY forest_test.csv .
COPY forest_train.csv .
COPY forest_val.csv .

View File

@ -1,8 +1,8 @@
name: mlflow_464914
conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
# docker_env:
# image: mlflow-docker-example-environment
# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
docker_env:
image: mlflow_image
entry_points:
main: