Files
mlflow--mlflow/examples/pytorch/torchscript/IrisClassification/MLproject
T
2026-07-13 13:22:34 +08:00

13 lines
221 B
Plaintext

name: iris-classification
python_env: python_env.yaml
entry_points:
main:
parameters:
epochs: {type: int, default: 50}
command: |
python iris_classification.py \
--epochs {epochs}