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)