name: s444380

docker_env:
    image: kamilguttmann/ium:mlflow

entry_points:
    main:
        parameters:
            epochs: {type: float, default: 10}
        command: "python3 train_model_mlflow.py {epochs}"
    test:
        command: "python3 evaluate_mlflow.py"