name: 434684-mlflow

docker_env:
    image: zollinka/ium:latest

entry_points:
    main:
        parameters:
            learning_rate: {type: float, default: 0.0001}
        command: "python3 ium_zadanie6_training.py {learning_rate}"