There was an error while loading. Please reload this page.
2 parents 63f685c + 006e32d commit 19dc7dfCopy full SHA for 19dc7df
fasttreeshap/explainers/_tree.py
@@ -1593,7 +1593,7 @@ def __init__(self, xgb_model):
1593
self.loss_chg.append(np.zeros(self.num_nodes[i], dtype=np.float32))
1594
self.sum_hess.append(np.zeros(self.num_nodes[i], dtype=np.float32))
1595
self.base_weight.append(np.zeros(self.num_nodes[i], dtype=np.float32))
1596
- self.leaf_child_cnt.append(np.zeros(self.num_nodes[i], dtype=np.int))
+ self.leaf_child_cnt.append(np.zeros(self.num_nodes[i], dtype=int))
1597
for j in range(self.num_nodes[i]):
1598
self.loss_chg[-1][j] = self.read('f')
1599
self.sum_hess[-1][j] = self.read('f')
0 commit comments