def setUp(self):
# pylint: disable=invalid-name
v1 = self.v1 = ContinuousVariable("v1")
v2 = self.v2 = DiscreteVariable("v2", "abc")
v3 = self.v3 = DiscreteVariable("v3", "def")
y = self.y = ContinuousVariable("y")
self.domain = Domain([v1, v2, v3], y)
self.data = Table(self.domain, np.arange(40).reshape(10, 4))
self.root = NumericNode(v1, 0, 13, np.array([0.0, 42]))
self.root.subset = np.array(np.arange(10), dtype=np.int32)
left = self.left = DiscreteNode(v2, 1, np.array([1, 42]))
left.subset = np.array([2, 3, 4, 5])
left.children = [Node(None, None, np.array([x, 42])) for x in [2, 3, 4]]
right = self.right = MappedDiscreteNode(
v3, 2, np.array([1, 1, 0]), np.array([5, 42])
)
right.children = [Node(None, None, np.array([6, 42])), None]
right.subset = np.array([8, 9])
self.root.children = [left, right]
self.model = TreeModel(self.data, self.root)
self.adapter = TreeAdapter(self.model)
def _get_tree_adapter(self, model):
if isinstance(model, SklModel):
return SklTreeAdapter(model)
return TreeAdapter(model)
def _get_tree_adapter(self, model):
return TreeAdapter(model)
