(self, booster, item_to_get, splitter)
| 23 | return bst |
| 24 | |
| 25 | def parse_dumped_model(self, booster, item_to_get, splitter): |
| 26 | item_to_get += "=" |
| 27 | txt_dump = booster.get_dump(with_stats=True) |
| 28 | tree_list = [tree.split("/n") for tree in txt_dump] |
| 29 | split_trees = [tree[0].split(item_to_get)[1:] for tree in tree_list] |
| 30 | res = sum( |
| 31 | [float(line.split(splitter)[0]) for tree in split_trees for line in tree] |
| 32 | ) |
| 33 | return res |
| 34 | |
| 35 | def test_trees_to_dataframe(self): |
| 36 | bst = self.build_model(max_depth=5, num_round=10) |
no test coverage detected