We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c7ec28f commit 1b0708eCopy full SHA for 1b0708e
1 file changed
vetiver/handlers/xgboost.py
@@ -11,12 +11,12 @@
11
12
13
class XGBoostHandler(BaseHandler):
14
- """Handler class for creating VetiverModels with statsmodels.
+ """Handler class for creating VetiverModels with xgboost.
15
16
Parameters
17
----------
18
- model : statsmodels
19
- a trained and fit statsmodels model
+ model :
+ a trained and fit xgboost model
20
"""
21
22
model_class = staticmethod(lambda: xgboost.Booster)
@@ -35,7 +35,7 @@ def create_meta(
35
url: str = None,
36
required_pkgs: list = [],
37
):
38
- """Create metadata for statsmodel"""
+ """Create metadata for xgboost"""
39
required_pkgs = required_pkgs + ["xgboost"]
40
meta = _model_meta(user, version, url, required_pkgs)
41
0 commit comments