flightrisk.models
=================

Risk track
----------

.. automodule:: flightrisk.models.risk.lightgbm_model
   :members:

.. automodule:: flightrisk.models.risk.xgboost_model
   :members:

.. automodule:: flightrisk.models.risk.calibration
   :members:

.. automodule:: flightrisk.models.risk.trainer
   :members:

Survival track
--------------

.. automodule:: flightrisk.models.survival.labels
   :members:

.. automodule:: flightrisk.models.survival.cox_model
   :members:

.. automodule:: flightrisk.models.survival.rsf_model
   :members:

.. automodule:: flightrisk.models.survival.trainer
   :members:

Uplift track
------------

.. automodule:: flightrisk.models.uplift.meta_learners
   :members:

.. automodule:: flightrisk.models.uplift.causal_forest
   :members:

.. automodule:: flightrisk.models.uplift.trainer
   :members: