ium_s437622/MLProject
2021-05-16 23:02:13 +02:00

11 lines
265 B
Plaintext

name: 437622-mlflow
docker_env:
image: jpogodzinski/ium:1
entry_points:
main:
parameters:
epochs: {type: int, default: 15}
batch_size: {type: int, default: 16}
command: "python3 zad8-mlflow.py {epochs} {batch_size}"