MLproject

name: Predicting flights cancellation

entry_points:
  main:
    command: "python3 -m src.main --multirun"
  evaluate:
    command: "python3 -m src.evaluate"
  validate:
    command: "python3 -m src.validate"
  transform:
    command: "python3 ./pipelines/data_prepare.py"
  extract:
    command: "./airflow.sh dags test data_extraction_v1_dag"
  deploy:
    command: "bash ./scripts/deploy_docker.sh"
  predict:
    parameters:
      example_version:
        type: string
        default: "v7.0"
      hostname:
        type: string
        default: "localhost"
      port:
        type: float
        default: 8083
      random_state:
        type: float
        default: 88
    command: "python3 -m src.predict ++example_version={example_version} ++hostname={hostname} ++port={port} ++random_state={random_state}"