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,