ium_444421/MLproject
2022-05-15 14:22:54 +02:00

12 lines
233 B
Plaintext

name: s444421
docker_env:
image: agakul/ium:mlflow
entry_points:
main:
parameters:
epochs: {type: float, default: 1000}
command: "python training_mlflow.py {epochs}"
test:
command: "python evaluation.py"