def get_ensemble_models(model, derived_col_names, col_names, target_name,
mining_imp_val, categoric_values):
"""
It returns the Mining Model element of the model
Parameters
----------
model :
Contains LGB model object.
derived_col_names : List
Contains column names after preprocessing.
col_names : List
Contains list of feature/column names.
target_name : String
Name of the Target column.
mining_imp_val : tuple
Contains the mining_attributes,mining_strategy, mining_impute_value.
categoric_values : tuple
Contains Categorical attribute names and its values
Returns
-------
mining_models :
Returns the MiningModel of the respective LGB model
"""
model_kwargs = sklToPmml.get_model_kwargs(model, col_names, target_name,
mining_imp_val)
mining_models = list()
mining_models.append(
pml.MiningModel(modelName="LightGBModel",
Segmentation=get_outer_segmentation(
model, derived_col_names, col_names, target_name,
mining_imp_val, categoric_values),
**model_kwargs))
return mining_models
def get_ensemble_models(model, derived_col_names, col_names, target_name, mining_imp_val,categoric_values,tasktype,modelPath):
"""
It returns the Mining Model element of the model
Parameters
----------
model :
Contains Xgboost model object.
derived_col_names : List
Contains column names after preprocessing.
col_names : List
Contains list of feature/column names.
target_name : String
Name of the Target column.
mining_imp_val : tuple
Contains the mining_attributes,mining_strategy, mining_impute_value.
categoric_values : tuple
Contains Categorical attribute names and its values
Returns
-------
mining_models :
Returns the MiningModel of the respective Xgboost model
"""
model_kwargs = skl_to_pmml.get_model_kwargs(model, col_names, target_name, mining_imp_val, categoric_values)
if 'XGBRegressor' in str(model.__class__):
model_kwargs['Targets'] = skl_to_pmml.get_targets(model, target_name)
mining_models = list()
mining_models.append(pml.MiningModel(
modelName="XGBoostModel",
Segmentation=get_outer_segmentation(model, col_names, col_names, target_name, mining_imp_val,categoric_values,tasktype,modelPath),
**model_kwargs
))
return mining_models
