name: 434804

docker_env: my_env.yaml
    image: s434804/ium:0.4

entry_points:
    main:
        parameters:
            test_size: {type: int, default: 0.2}
            epochs: {type: int, default: 100}
            batch_size: {type: int, default: 32}
        command: "python3 mlflow.py {test_size} {epochs} {batch_size}"