15 lines
315 B
YAML
15 lines
315 B
YAML
|
name: OrangeQualityModel
|
||
|
|
||
|
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}"
|