Skip to content

GitLab

Commit d5bbaa26 authored by wxchan's avatar wxchan Committed by Qiwei Ye
Browse files

fix getstate in _InnerPredictor (#366)

parent 0dbc421a
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
python-package/lightgbm/basic.py
View file @ d5bbaa26
...@@ -343,6 +343,11 @@ class _InnerPredictor(object): ...@@ -343,6 +343,11 @@ class _InnerPredictor(object):
if self.__is_manage_handle: if self.__is_manage_handle:
_safe_call(_LIB.LGBM_BoosterFree(self.handle)) _safe_call(_LIB.LGBM_BoosterFree(self.handle))
def __getstate__(self):
this = self.__dict__.copy()
this.pop('handle', None)
return this
def predict(self, data, num_iteration=-1, def predict(self, data, num_iteration=-1,
raw_score=False, pred_leaf=False, data_has_header=False, raw_score=False, pred_leaf=False, data_has_header=False,
is_reshape=True): is_reshape=True):
......
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