"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "acb677416af59a5cc65c8dff7adfb8e020842d46"
Commit 9bb3b0d0 authored by Guolin Ke's avatar Guolin Ke
Browse files

[python] add get_leaf_output in Booster

parent bd4c4066
...@@ -1618,6 +1618,15 @@ class Booster(object): ...@@ -1618,6 +1618,15 @@ class Booster(object):
num_iteration = self.best_iteration num_iteration = self.best_iteration
return predictor.predict(data, num_iteration, raw_score, pred_leaf, data_has_header, is_reshape) return predictor.predict(data, num_iteration, raw_score, pred_leaf, data_has_header, is_reshape)
def get_leaf_output(tree_id, leaf_id):
ret = ctypes.c_double(0)
_safe_call(_LIB.LGBM_BoosterGetLeafValue(
self.handle,
ctypes.c_int(tree_id),
ctypes.c_int(leaf_id),
ctypes.byref(ret)))
return ret.value
def _to_predictor(self, pred_parameter=None): def _to_predictor(self, pred_parameter=None):
"""Convert to predictor""" """Convert to predictor"""
predictor = _InnerPredictor(booster_handle=self.handle, pred_parameter=pred_parameter) predictor = _InnerPredictor(booster_handle=self.handle, pred_parameter=pred_parameter)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment