We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 74dc571 commit 12f7db2Copy full SHA for 12f7db2
unboxapi/models.py
@@ -25,6 +25,8 @@ class ModelType(Enum):
25
pytorch = "PytorchModelArtifact"
26
#: For models built with `TensorFlow <https://www.tensorflow.org/>`_.
27
tensorflow = "TensorflowSavedModelArtifact"
28
+ #: For models built with `XGBoost <https://xgboost.readthedocs.io>`_.
29
+ xgboost = "XgboostModelArtifact"
30
#: For models built with `Hugging Face transformers <https://huggingface.co/docs/transformers/index>`_.
31
transformers = "TransformersModelArtifact"
32
#: For models built with `Keras <https://keras.io/>`_.
0 commit comments