(self, X, node, prob=False)
| 203 | return ig |
| 204 | |
| 205 | def _traverse(self, X, node, prob=False): |
| 206 | if isinstance(node, Leaf): |
| 207 | if self.classifier: |
| 208 | return node.value if prob else node.value.argmax() |
| 209 | return node.value |
| 210 | if X[node.feature] <= node.threshold: |
| 211 | return self._traverse(X, node.left, prob) |
| 212 | return self._traverse(X, node.right, prob) |
| 213 | |
| 214 | |
| 215 | def mse(y): |
no outgoing calls
no test coverage detected