ium_464914/mlflow/MLProject
Alicja Szulecka ed9927d7a1 mlflow
2024-05-06 17:27:28 +02:00

13 lines
331 B
Plaintext

name: mlflow_464914
conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
# docker_env:
# image: mlflow-docker-example-environment
entry_points:
main:
parameters:
epochs: {type: int, default: 10}
command: "python mlflow_model.py {epochs}"
test:
command: "python mlflow_prediction.py"