From 464421f7b003ed4930f21807db15e20228c54592 Mon Sep 17 00:00:00 2001 From: Microbotz Date: Sat, 18 Mar 2023 09:42:45 -0700 Subject: [PATCH] Add Support for XGBRanker --- m2cgen/assemblers/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/m2cgen/assemblers/__init__.py b/m2cgen/assemblers/__init__.py index a817d331..4c652b7b 100644 --- a/m2cgen/assemblers/__init__.py +++ b/m2cgen/assemblers/__init__.py @@ -46,6 +46,7 @@ "xgboost_XGBRFClassifier": XGBoostModelAssemblerSelector, "xgboost_XGBRegressor": XGBoostModelAssemblerSelector, "xgboost_XGBRFRegressor": XGBoostModelAssemblerSelector, + "xgboost_XGBRanker": XGBoostModelAssemblerSelector, # Sklearn SVM "sklearn_LinearSVC": SklearnLinearModelAssembler,