From 38ae164e25fb17a6d5ddfe3a13b3aa6101de1754 Mon Sep 17 00:00:00 2001 From: Daniel Potapov Date: Thu, 7 Dec 2017 13:58:50 +0300 Subject: [PATCH] Fix GH-248 --- eli5/xgboost.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/eli5/xgboost.py b/eli5/xgboost.py index f565d9c8..8a3d0f5d 100644 --- a/eli5/xgboost.py +++ b/eli5/xgboost.py @@ -356,18 +356,19 @@ def _parse_tree_dump(text_dump): def _parse_dump_line(line): # type: (str) -> Tuple[int, Dict[str, Any]] branch_match = re.match( - '^(\t*)(\d+):\[([^<]+)<([^\]]+)\] ' + '^(\t*)(\d+):\[([^<]+)