ium_464913/mlflow/MLproject
2024-05-12 12:14:43 +02:00

11 lines
270 B
Plaintext

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}'