name: 434742-mlflow

docker_env:
    image: patlaz/ium:1.0

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