2024-05-26 14:26:09 +02:00
|
|
|
name: mlflow_464906
|
2024-05-26 14:16:50 +02:00
|
|
|
|
|
|
|
|
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}"
|
