2024-05-06 17:27:28 +02:00
|
|
|
name: mlflow_464914
|
|
|
|
|
|
2024-05-06 22:08:05 +02:00
|
|
|
# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
|
|
|
|
|
docker_env:
|
|
|
|
|
image: mlflow_image
|
2024-05-06 17:27:28 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: int, default: 10}
|
|
|
|
|
command: "python mlflow_model.py {epochs}"
|
|
|
|
|
test:
|
|
|
|
|
command: "python mlflow_prediction.py"
|
