name: HousePriceModel

conda_env: environment.yml

entry_points:
  main:
    parameters:
      epochs: {type: int, default: 20}
      learning_rate: {type: float, default: 0.001}
      batch_size: {type: int, default: 32}
    command: "python create_model.py {epochs} {learning_rate} {batch_size}"

  test:
    command: "python evaluate.py"