name: Credit card fraud MLFlow - s464913

conda_env: conda.yaml

entry_points:
  main:
    parameters:
      learning_rate: { type: float, default: 0.001 }
      epochs: { type: int, default: 5 }
    command: 'python mlflow_train_evaluation.py {learning_rate} {epochs}'