def get_outer_segmentation(model, derived_col_names, col_names, target_name,
mining_imp_val, categoric_values, model_name):
"""
It returns the Segmentation 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
model_name : string
Name of the model
Returns
-------
segmentation :
Get the outer most Segmentation of an LGB model
"""
if 'LGBMRegressor' in str(model.__class__):
segmentation = get_segments(model, derived_col_names, col_names,
target_name, mining_imp_val,
categoric_values, model_name)
else:
segmentation = pml.Segmentation(
multipleModelMethod=get_multiple_model_method(model),
Segment=get_segments(model, derived_col_names, col_names,
target_name, mining_imp_val, categoric_values,
model_name))
return segmentation
def get_segments_for_lgbr(model, derived_col_names, feature_names, target_name,
mining_imp_val, categorical_values):
"""
It returns all the Segments element of the model
Parameters
----------
model :
Contains LGB model object.
derived_col_names : List
Contains column names after preprocessing.
feature_names : List
Contains list of feature/column names.
target_name : List
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
-------
segment :
Get the Segmentation element which contains inner segments.
"""
segments = list()
main_key_value = []
lgb_dump = model.booster_.dump_model()
for i in range(len(lgb_dump['tree_info'])):
tree = lgb_dump['tree_info'][i]['tree_structure']
main_key_value.append(tree)
segmentation = pml.Segmentation(
multipleModelMethod=MULTIPLE_MODEL_METHOD.SUM,
Segment=generate_Segments_Equal_To_Estimators(main_key_value,
derived_col_names,
feature_names))
return segmentation
