name: 426206mlflow

#conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
    
#docker_env:
#  image: mlflow-docker-example-environment

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