Skip to content

Commit 12f7db2

Browse files
gustavocidornelaswhoseoyster
authored andcommitted
Completes UNB-2253 - XGBoost native support
1 parent 74dc571 commit 12f7db2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

unboxapi/models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ class ModelType(Enum):
2525
pytorch = "PytorchModelArtifact"
2626
#: For models built with `TensorFlow <https://www.tensorflow.org/>`_.
2727
tensorflow = "TensorflowSavedModelArtifact"
28+
#: For models built with `XGBoost <https://xgboost.readthedocs.io>`_.
29+
xgboost = "XgboostModelArtifact"
2830
#: For models built with `Hugging Face transformers <https://huggingface.co/docs/transformers/index>`_.
2931
transformers = "TransformersModelArtifact"
3032
#: For models built with `Keras <https://keras.io/>`_.

0 commit comments

Comments
 (0)