name: 434695-mlflow

docker_env:
    image: shroomy/ium:1

entry_points:
    main:
        parameters:
            epochs: {type: int, default: 15}
            batch_size: {type: int, default: 16}
        command: "python3 vgsales-mlflow.py {epochs} {batch_size}"