15 lines
310 B
YAML
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}"
|