name: 434695-mlflow
docker_env:
image: shroomy/ium:7
entry_points:
main:
parameters:
epochs: {type: int, default: 15}
batch_size: {type: int, default: 16}
command: "python3 vgsales-mlflow.py {epochs} {batch_size}"
