ium_464962/mlflow/MLProject

13 lines
282 B
Plaintext
Raw Normal View History

2024-05-21 19:47:48 +02:00
name: Car Price Prediction
conda_env: conda.yaml
entry_points:
main:
parameters:
epochs: {type: int, default: 20}
batch_size: {type: int, default: 32}
command: "python mlflow_model.py {epochs} {batch_size}"
predict:
command: "python mlflow_predict.py"