| 174 | |
| 175 | |
| 176 | class BallTreeNode: |
| 177 | def __init__(self, centroid=None, X=None, y=None): |
| 178 | self.left = None |
| 179 | self.right = None |
| 180 | self.radius = None |
| 181 | self.is_leaf = False |
| 182 | |
| 183 | self.data = X |
| 184 | self.targets = y |
| 185 | self.centroid = centroid |
| 186 | |
| 187 | def __repr__(self): |
| 188 | fstr = "BallTreeNode(centroid={}, is_leaf={})" |
| 189 | return fstr.format(self.centroid, self.is_leaf) |
| 190 | |
| 191 | def to_dict(self): |
| 192 | d = self.__dict__ |
| 193 | d["id"] = "BallTreeNode" |
| 194 | return d |
| 195 | |
| 196 | |
| 197 | class BallTree: |
no outgoing calls
no test coverage detected