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):
num_iteration = self.best_iteration
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):
"""Convert to predictor"""
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