2024-05-21 19:49:10 +02:00
|
|
|
name: mlflow_464962
|
2024-05-21 19:47:48 +02:00
|
|
|
|
|
|
|
|
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"
|
