ium_464979/mlflow_project/MLproject
AWieczarek d739f275e8 IUM_08
2024-05-13 20:25:54 +02:00

11 lines
252 B
Plaintext

name: MLflow_s464979
conda_env: conda.yaml
entry_points:
optimal_parameters:
parameters:
epochs: { type: int, default: 20 }
batch_size: { type: int, default: 32 }
command: 'python mlflow_training_model.py {epochs} {batch_size}'