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}"