name: 434788-mlflow

docker_env:
    image: snowycocoon/ium_434788:4

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