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}"