fixed missing part

89444e57 · Dr.李 · 0858d3cb · 89444e57
Commit 89444e57 authored Mar 31, 2019 by Dr.李
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

treemodel.py alphamind/model/treemodel.py +1 -0

No files found.
--- a/alphamind/model/treemodel.py
+++ b/alphamind/model/treemodel.py
@@ -71,6 +71,7 @@ class XGBRegressor(create_model_base('xgboost')):
                                     n_jobs=n_jobs,
                                     missing=missing,
                                     **kwargs)
+        self.impl = XGBRegressor.model_decode(self.model_encode())

    @property
    def importances(self):