name: 426206mlflow

#conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
    
docker_env:
  image: rokoch/ium:01

entry_points:
  main:
    parameters:
      epochs: {type: int, default: 100}
      lr: {type: float, default: 0.001}
    command: "python3 train_mlflow.py -e {epochs} -l {lr}"