mirror of
https://github.com/vale981/ray
synced 2025-03-10 05:16:49 -04:00
128f9e5664
As the integration logging callbacks are commonly used with AIR Trainers, they should be moved from the tune package to the air package. The old imports will still work, but raise a deprecation warning.
81 lines
1.9 KiB
ReStructuredText
81 lines
1.9 KiB
ReStructuredText
.. _tune-integration:
|
|
|
|
External library integrations (tune.integration)
|
|
================================================
|
|
|
|
.. contents::
|
|
:local:
|
|
:depth: 1
|
|
|
|
|
|
.. _tune-integration-keras:
|
|
|
|
Keras (tune.integration.keras)
|
|
------------------------------------------------------
|
|
|
|
.. autoclass:: ray.tune.integration.keras.TuneReportCallback
|
|
|
|
.. autoclass:: ray.tune.integration.keras.TuneReportCheckpointCallback
|
|
|
|
|
|
.. _tune-integration-mlflow:
|
|
|
|
MLflow (tune.integration.mlflow)
|
|
--------------------------------
|
|
|
|
:ref:`See also here <tune-mlflow-ref>`.
|
|
|
|
.. autoclass:: ray.air.callbacks.mlflow.MLflowLoggerCallback
|
|
|
|
.. autofunction:: ray.tune.integration.mlflow.mlflow_mixin
|
|
|
|
|
|
.. _tune-integration-mxnet:
|
|
|
|
MXNet (tune.integration.mxnet)
|
|
------------------------------
|
|
|
|
.. autoclass:: ray.tune.integration.mxnet.TuneReportCallback
|
|
|
|
.. autoclass:: ray.tune.integration.mxnet.TuneCheckpointCallback
|
|
|
|
|
|
.. _tune-integration-pytorch-lightning:
|
|
|
|
PyTorch Lightning (tune.integration.pytorch_lightning)
|
|
------------------------------------------------------
|
|
|
|
.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCallback
|
|
|
|
.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCheckpointCallback
|
|
|
|
.. _tune-integration-wandb:
|
|
|
|
Weights and Biases (tune.integration.wandb)
|
|
-------------------------------------------
|
|
|
|
:ref:`See also here <tune-wandb-ref>`.
|
|
|
|
.. autoclass:: ray.air.callbacks.wandb.WandbLoggerCallback
|
|
|
|
.. autofunction:: ray.tune.integration.wandb.wandb_mixin
|
|
|
|
|
|
.. _tune-integration-xgboost:
|
|
|
|
XGBoost (tune.integration.xgboost)
|
|
----------------------------------
|
|
|
|
.. autoclass:: ray.tune.integration.xgboost.TuneReportCallback
|
|
|
|
.. autoclass:: ray.tune.integration.xgboost.TuneReportCheckpointCallback
|
|
|
|
|
|
.. _tune-integration-lightgbm:
|
|
|
|
LightGBM (tune.integration.lightgbm)
|
|
------------------------------------
|
|
|
|
.. autoclass:: ray.tune.integration.lightgbm.TuneReportCallback
|
|
|
|
.. autoclass:: ray.tune.integration.lightgbm.TuneReportCheckpointCallback
|