ium_464906/MLProject.yaml
zgolebiewska e5af4f2f39 IUM_08
2024-05-26 14:26:09 +02:00

15 lines
310 B
YAML

name: mlflow_464906
conda_env: conda.yaml
entry_points:
train:
parameters:
epochs: {type: int, default: 100}
command: "python model.py"
test:
parameters:
model_path: {type: str, default: "orange_quality_model_tf.h5"}
command: "python test_model.py --model_path {model_path}"