name: s444421
    
docker_env:
  image: agakul/ium:mlflow

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 1000}
    command: "python training_mlflow.py {epochs}"
  test:
    command: "python evaluation.py"