name: iris-classification

conda_env: conda.yaml

entry_points:
  main:
    parameters:
      epochs: {type: int, default: 50}

    command: |
          python iris_classification.py \
            --epochs {epochs}
