def _draw_trees(self):
self.grid_items, self.ptrees = [], []
num_trees = len(self.forest_adapter.get_trees())
with self.progressBar(num_trees) as prg, self.disable_ui():
for tree in self.forest_adapter.get_trees():
ptree = PythagorasTreeViewer(
None,
tree,
interactive=False,
padding=100,
target_class_index=self.target_class_index,
weight_adjustment=self.SIZE_CALCULATION[
self.size_calc_idx][1])
grid_item = GridItem(ptree,
self.grid,
max_size=self._calculate_zoom(self.zoom))
# We don't want to show flickering while the trees are being
grid_item.setVisible(False)
self.grid_items.append(grid_item)
self.ptrees.append(ptree)
prg.advance()
self.grid.set_items(self.grid_items)
# This is necessary when adding items for the first time
if self.grid:
width = (self.view.width() -
self.view.verticalScrollBar().width())
self.grid.reflow(width)
self.grid.setPreferredWidth(width)
# After drawing is complete, we show the trees
for grid_item in self.grid_items:
grid_item.setVisible(True)
def _draw_trees(self):
self.ui_size_calc_combo.setEnabled(False)
self.grid_items, self.ptrees = [], []
with self.progressBar(len(self.forest_adapter.get_trees())) as prg:
for tree in self.forest_adapter.get_trees():
ptree = PythagorasTreeViewer(
None,
tree,
node_color_func=self._type_specific('_get_node_color'),
interactive=False,
padding=100)
self.grid_items.append(
GridItem(ptree,
self.grid,
max_size=self._calculate_zoom(self.zoom)))
self.ptrees.append(ptree)
prg.advance()
self.grid.set_items(self.grid_items)
# This is necessary when adding items for the first time
if self.grid:
width = (self.view.width() - self.view.verticalScrollBar().width())
self.grid.reflow(width)
self.grid.setPreferredWidth(width)
self.ui_size_calc_combo.setEnabled(True)
def data(self, index, role=Qt.DisplayRole):
# type: (QModelIndex, Qt.QDisplayRole) -> Any
if not index.isValid():
return None
idx = index.row()
if role == Qt.SizeHintRole:
return self.item_scale * QSize(100, 100)
if role == Qt.DisplayRole:
if 'tree' not in self._other_data[idx]:
scene = QGraphicsScene(parent=self)
tree = PythagorasTreeViewer(
adapter=self._list[idx],
weight_adjustment=OWPythagoreanForest.SIZE_CALCULATION[
self.size_calc_idx][1],
interactive=False,
padding=100,
depth_limit=self.depth_limit,
target_class_index=self.target_class_idx,
)
scene.addItem(tree)
self._other_data[idx]['scene'] = scene
self._other_data[idx]['tree'] = tree
return self._other_data[idx]['scene']
return super().data(index, role)
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.info = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(box_display,
self,
'depth_limit',
label='Depth',
ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(box_display,
self,
'target_class_index',
label='Target class',
orientation=Qt.Horizontal,
items=[],
contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display,
self,
'size_calc_idx',
label='Size',
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[0],
contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(box_display,
self,
'size_log_scale',
label='Log scale factor',
minValue=1,
maxValue=100,
ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
self.cb_show_tooltips = gui.checkBox(
box_plot,
self,
'tooltips_enabled',
label='Enable tooltips',
callback=self.update_tooltip_enabled)
self.cb_show_legend = gui.checkBox(box_plot,
self,
'show_legend',
label='Show legend',
callback=self.update_show_legend)
gui.button(self.controlArea,
self,
label="Redraw",
callback=self.redraw)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
class OWPythagorasTree(OWWidget):
name = '毕达哥拉斯树'
description = '毕达哥拉斯的树状结构可视化'
icon = 'icons/PythagoreanTree.svg'
keywords = ["fractal"]
priority = 1000
class Inputs:
tree = Input("Tree", TreeModel)
class Outputs:
selected_data = Output("Selected Data", Table, default=True)
annotated_data = Output(ANNOTATED_DATA_SIGNAL_NAME, Table)
# Enable the save as feature
graph_name = '场景'
# Settings
depth_limit = settings.ContextSetting(10)
target_class_index = settings.ContextSetting(0)
size_calc_idx = settings.Setting(0)
size_log_scale = settings.Setting(2)
tooltips_enabled = settings.Setting(True)
show_legend = settings.Setting(False)
LEGEND_OPTIONS = {
'corner': Anchorable.BOTTOM_RIGHT,
'offset': (10, 10),
}
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.info = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(box_display,
self,
'depth_limit',
label='Depth',
ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(box_display,
self,
'target_class_index',
label='Target class',
orientation=Qt.Horizontal,
items=[],
contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display,
self,
'size_calc_idx',
label='Size',
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[0],
contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(box_display,
self,
'size_log_scale',
label='Log scale factor',
minValue=1,
maxValue=100,
ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
self.cb_show_tooltips = gui.checkBox(
box_plot,
self,
'tooltips_enabled',
label='Enable tooltips',
callback=self.update_tooltip_enabled)
self.cb_show_legend = gui.checkBox(box_plot,
self,
'show_legend',
label='Show legend',
callback=self.update_show_legend)
gui.button(self.controlArea,
self,
label="Redraw",
callback=self.redraw)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
@Inputs.tree
def set_tree(self, model=None):
"""When a different tree is given."""
self.clear()
self.model = model
if model is not None:
self.instances = model.instances
# this bit is important for the regression classifier
if self.instances is not None and \
self.instances.domain != model.domain:
self.clf_dataset = self.instances.transform(self.model.domain)
else:
self.clf_dataset = self.instances
self.tree_adapter = self._get_tree_adapter(self.model)
self.ptree.clear()
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self._update_depth_slider()
self.color_palette = self.ptree.root.color_palette
self._update_legend_colors()
self._update_legend_visibility()
self._update_info_box()
self._update_target_class_combo()
self._update_main_area()
# The target class can also be passed from the meta properties
# This must be set after `_update_target_class_combo`
if hasattr(model, 'meta_target_class_index'):
self.target_class_index = model.meta_target_class_index
self.update_colors()
# Get meta variables describing what the settings should look like
# if the tree is passed from the Pythagorean forest widget.
if hasattr(model, 'meta_size_calc_idx'):
self.size_calc_idx = model.meta_size_calc_idx
self.update_size_calc()
# TODO There is still something wrong with this
# if hasattr(model, 'meta_depth_limit'):
# self.depth_limit = model.meta_depth_limit
# self.update_depth()
self.Outputs.annotated_data.send(
create_annotated_table(self.instances, None))
def clear(self):
"""Clear all relevant data from the widget."""
self.model = None
self.instances = None
self.clf_dataset = None
self.tree_adapter = None
if self.legend is not None:
self.scene.removeItem(self.legend)
self.legend = None
self.ptree.clear()
self._clear_info_box()
self._clear_target_class_combo()
self._clear_depth_slider()
self._update_log_scale_slider()
def update_depth(self):
"""This method should be called when the depth changes"""
self.ptree.set_depth_limit(self.depth_limit)
def update_colors(self):
"""When the target class / node coloring needs to be updated."""
self.ptree.target_class_changed(self.target_class_index)
self._update_legend_colors()
def update_size_calc(self):
"""When the tree size calculation is updated."""
self._update_log_scale_slider()
self.invalidate_tree()
def redraw(self):
self.tree_adapter.shuffle_children()
self.invalidate_tree()
def invalidate_tree(self):
"""When the tree needs to be completely recalculated."""
if self.model is not None:
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self.ptree.set_depth_limit(self.depth_limit)
self._update_main_area()
def update_tooltip_enabled(self):
"""When the tooltip visibility is changed and need to be updated."""
self.ptree.tooltip_changed(self.tooltips_enabled)
def update_show_legend(self):
"""When the legend visibility needs to be updated."""
self._update_legend_visibility()
def _update_info_box(self):
self.info.setText('Nodes: {}\nDepth: {}'.format(
self.tree_adapter.num_nodes, self.tree_adapter.max_depth))
def _update_depth_slider(self):
self.depth_slider.parent().setEnabled(True)
self.depth_slider.setMaximum(self.tree_adapter.max_depth)
self._set_max_depth()
def _update_legend_visibility(self):
if self.legend is not None:
self.legend.setVisible(self.show_legend)
def _update_log_scale_slider(self):
"""On calc method combo box changed."""
self.log_scale_box.parent().setEnabled(
self.SIZE_CALCULATION[self.size_calc_idx][0] == 'Logarithmic')
def _clear_info_box(self):
self.info.setText('没有树输入')
def _clear_depth_slider(self):
self.depth_slider.parent().setEnabled(False)
self.depth_slider.setMaximum(0)
def _clear_target_class_combo(self):
self.target_class_combo.clear()
self.target_class_index = 0
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _set_max_depth(self):
"""Set the depth to the max depth and update appropriate actors."""
self.depth_limit = self.tree_adapter.max_depth
self.depth_slider.setValue(self.depth_limit)
def _update_main_area(self):
# refresh the scene rect, cuts away the excess whitespace, and adds
# padding for panning.
self.scene.setSceneRect(self.view.central_widget_rect())
# reset the zoom level
self.view.recalculate_and_fit()
self.view.update_anchored_items()
def _get_tree_adapter(self, model):
if isinstance(model, SklModel):
return SklTreeAdapter(model)
return TreeAdapter(model)
def onDeleteWidget(self):
"""When deleting the widget."""
super().onDeleteWidget()
self.clear()
def commit(self):
"""Commit the selected data to output."""
if self.instances is None:
self.Outputs.selected_data.send(None)
self.Outputs.annotated_data.send(None)
return
nodes = [
i.tree_node.label for i in self.scene.selectedItems()
if isinstance(i, SquareGraphicsItem)
]
data = self.tree_adapter.get_instances_in_nodes(nodes)
self.Outputs.selected_data.send(data)
selected_indices = self.tree_adapter.get_indices(nodes)
self.Outputs.annotated_data.send(
create_annotated_table(self.instances, selected_indices))
def send_report(self):
"""Send report."""
self.report_plot()
def _update_target_class_combo(self):
self._clear_target_class_combo()
label = [
x for x in self.target_class_combo.parent().children()
if isinstance(x, QLabel)
][0]
if self.instances.domain.has_discrete_class:
label_text = '目标类'
values = [
c.title() for c in self.instances.domain.class_vars[0].values
]
values.insert(0, 'None')
else:
label_text = '节点颜色'
values = list(ContinuousTreeNode.COLOR_METHODS.keys())
label.setText(label_text)
self.target_class_combo.addItems(values)
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
if self.instances.domain.has_discrete_class:
self._classification_update_legend_colors()
else:
self._regression_update_legend_colors()
def _classification_update_legend_colors(self):
if self.target_class_index == 0:
self.legend = OWDiscreteLegend(domain=self.model.domain,
**self.LEGEND_OPTIONS)
else:
items = ((self.target_class_combo.itemText(
self.target_class_index),
self.color_palette[self.target_class_index - 1]),
('other', QColor('#ffffff')))
self.legend = OWDiscreteLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _regression_update_legend_colors(self):
def _get_colors_domain(domain):
class_var = domain.class_var
start, end, pass_through_black = class_var.colors
if pass_through_black:
lst_colors = [QColor(*c) for c in [start, (0, 0, 0), end]]
else:
lst_colors = [QColor(*c) for c in [start, end]]
return lst_colors
# The colors are the class mean
if self.target_class_index == 1:
values = (np.min(self.clf_dataset.Y), np.max(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
items = list(zip(values, colors))
# Colors are the stddev
elif self.target_class_index == 2:
values = (0, np.std(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
items = list(zip(values, colors))
else:
items = None
self.legend = OWContinuousLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
class OWPythagorasTree(OWWidget):
name = '毕达哥拉斯树(Pythagorean Tree)'
description = '类似树结构的毕达哥拉斯树可视化。'
icon = 'icons/PythagoreanTree.svg'
keywords = ["fractal"]
priority = 1000
class Inputs:
tree = Input("树(Tree)", TreeModel, replaces=['Tree'])
class Outputs:
selected_data = Output("选定的数据(Selected Data)",
Table,
default=True,
replaces=['Selected Data'])
annotated_data = Output(ANNOTATED_DATA_SIGNAL_Chinese_NAME,
Table,
replaces=['Data'])
# Enable the save as feature
graph_name = 'scene'
# Settings
settingsHandler = settings.DomainContextHandler()
depth_limit = settings.ContextSetting(10)
target_class_index = settings.ContextSetting(0)
size_calc_idx = settings.Setting(0)
size_log_scale = settings.Setting(2)
tooltips_enabled = settings.Setting(True)
show_legend = settings.Setting(False)
LEGEND_OPTIONS = {
'corner': Anchorable.BOTTOM_RIGHT,
'offset': (10, 10),
}
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.data = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('正常', lambda x: x),
('平方根', lambda x: sqrt(x)),
('对数的', lambda x: log(x * self.size_log_scale + 1)),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, '树信息')
self.infolabel = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, '显示设置')
self.depth_slider = gui.hSlider(box_display,
self,
'depth_limit',
label='深度',
ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(box_display,
self,
'target_class_index',
label='目标类别',
orientation=Qt.Horizontal,
items=[],
contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display,
self,
'size_calc_idx',
label='大小',
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[0],
contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(box_display,
self,
'size_log_scale',
label='对数比例因子',
minValue=1,
maxValue=100,
ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, '绘图属性')
self.cb_show_tooltips = gui.checkBox(
box_plot,
self,
'tooltips_enabled',
label='启动工具提示',
callback=self.update_tooltip_enabled)
self.cb_show_legend = gui.checkBox(box_plot,
self,
'show_legend',
label='显示图例',
callback=self.update_show_legend)
gui.button(self.controlArea, self, label="重新绘制", callback=self.redraw)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
@Inputs.tree
def set_tree(self, model=None):
"""When a different tree is given."""
self.closeContext()
self.clear()
self.model = model
if model is not None:
self.data = model.instances
self.tree_adapter = self._get_tree_adapter(self.model)
self.ptree.clear()
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self._update_depth_slider()
self.color_palette = self.ptree.root.color_palette
self._update_legend_colors()
self._update_legend_visibility()
self._update_info_box()
self._update_target_class_combo()
self._update_main_area()
self.openContext(self.model)
self.update_depth()
# The forest widget sets the following attributes on the tree,
# describing the settings on the forest widget. To keep the tree
# looking the same as on the forest widget, we prefer these settings to
# context settings, if set.
if hasattr(model, "meta_target_class_index"):
self.target_class_index = model.meta_target_class_index
self.update_colors()
if hasattr(model, "meta_size_calc_idx"):
self.size_calc_idx = model.meta_size_calc_idx
self.update_size_calc()
if hasattr(model, "meta_depth_limit"):
self.depth_limit = model.meta_depth_limit
self.update_depth()
self.Outputs.annotated_data.send(
create_annotated_table(self.data, None))
def clear(self):
"""Clear all relevant data from the widget."""
self.model = None
self.data = None
self.tree_adapter = None
if self.legend is not None:
self.scene.removeItem(self.legend)
self.legend = None
self.ptree.clear()
self._clear_info_box()
self._clear_target_class_combo()
self._clear_depth_slider()
self._update_log_scale_slider()
def update_depth(self):
"""This method should be called when the depth changes"""
self.ptree.set_depth_limit(self.depth_limit)
def update_colors(self):
"""When the target class / node coloring needs to be updated."""
self.ptree.target_class_changed(self.target_class_index)
self._update_legend_colors()
def update_size_calc(self):
"""When the tree size calculation is updated."""
self._update_log_scale_slider()
self.invalidate_tree()
def redraw(self):
if self.data is None:
return
self.tree_adapter.shuffle_children()
self.invalidate_tree()
def invalidate_tree(self):
"""When the tree needs to be completely recalculated."""
if self.model is not None:
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self.ptree.set_depth_limit(self.depth_limit)
self._update_main_area()
def update_tooltip_enabled(self):
"""When the tooltip visibility is changed and need to be updated."""
self.ptree.tooltip_changed(self.tooltips_enabled)
def update_show_legend(self):
"""When the legend visibility needs to be updated."""
self._update_legend_visibility()
def _update_info_box(self):
self.infolabel.setText('节点: {}\n深度: {}'.format(
self.tree_adapter.num_nodes, self.tree_adapter.max_depth))
def _update_depth_slider(self):
self.depth_slider.parent().setEnabled(True)
self.depth_slider.setMaximum(self.tree_adapter.max_depth)
self._set_max_depth()
def _update_legend_visibility(self):
if self.legend is not None:
self.legend.setVisible(self.show_legend)
def _update_log_scale_slider(self):
"""On calc method combo box changed."""
self.log_scale_box.parent().setEnabled(
self.SIZE_CALCULATION[self.size_calc_idx][0] == '对数的')
def _clear_info_box(self):
self.infolabel.setText('没有树输入')
def _clear_depth_slider(self):
self.depth_slider.parent().setEnabled(False)
self.depth_slider.setMaximum(0)
def _clear_target_class_combo(self):
self.target_class_combo.clear()
self.target_class_index = 0
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _set_max_depth(self):
"""Set the depth to the max depth and update appropriate actors."""
self.depth_limit = self.tree_adapter.max_depth
self.depth_slider.setValue(self.depth_limit)
def _update_main_area(self):
# refresh the scene rect, cuts away the excess whitespace, and adds
# padding for panning.
self.scene.setSceneRect(self.view.central_widget_rect())
# reset the zoom level
self.view.recalculate_and_fit()
self.view.update_anchored_items()
def _get_tree_adapter(self, model):
if isinstance(model, SklModel):
return SklTreeAdapter(model)
return TreeAdapter(model)
def onDeleteWidget(self):
"""When deleting the widget."""
super().onDeleteWidget()
self.clear()
def commit(self):
"""Commit the selected data to output."""
if self.data is None:
self.Outputs.selected_data.send(None)
self.Outputs.annotated_data.send(None)
return
nodes = [
i.tree_node.label for i in self.scene.selectedItems()
if isinstance(i, SquareGraphicsItem)
]
data = self.tree_adapter.get_instances_in_nodes(nodes)
self.Outputs.selected_data.send(data)
selected_indices = self.tree_adapter.get_indices(nodes)
self.Outputs.annotated_data.send(
create_annotated_table(self.data, selected_indices))
def send_report(self):
"""Send report."""
self.report_plot()
def _update_target_class_combo(self):
self._clear_target_class_combo()
label = [
x for x in self.target_class_combo.parent().children()
if isinstance(x, QLabel)
][0]
if self.data.domain.has_discrete_class:
label_text = '目标类别'
values = [c.title() for c in self.data.domain.class_vars[0].values]
values.insert(0, 'None')
else:
label_text = 'Node color'
values = list(ContinuousTreeNode.COLOR_METHODS.keys())
label.setText(label_text)
self.target_class_combo.addItems(values)
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
if self.data.domain.has_discrete_class:
self._classification_update_legend_colors()
else:
self._regression_update_legend_colors()
def _classification_update_legend_colors(self):
if self.target_class_index == 0:
self.legend = OWDiscreteLegend(domain=self.model.domain,
**self.LEGEND_OPTIONS)
else:
items = ((self.target_class_combo.itemText(
self.target_class_index),
self.color_palette[self.target_class_index - 1]),
('other', QColor('#ffffff')))
self.legend = OWDiscreteLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _regression_update_legend_colors(self):
# The colors are the class mean
palette = self.model.domain.class_var.palette
if self.target_class_index == 1:
items = ((np.min(self.data.Y), np.max(self.data.Y)), palette)
# Colors are the stddev
elif self.target_class_index == 2:
items = ((0, np.std(self.data.Y)), palette)
else:
items = None
self.legend = OWContinuousLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def __init__(self):
super().__init__()
# Instance variables
self.tree_type = self.GENERAL
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
# The +1 is there so that we don't get division by 0 exceptions
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# Color modes for regression trees
self.REGRESSION_COLOR_CALC = [
('None', lambda _, __: QtGui.QColor(255, 255, 255)),
('Class mean', self._color_class_mean),
('Standard deviation', self._color_stddev),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.info = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(
box_display, self, 'depth_limit', label='Depth', ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(
box_display, self, 'target_class_index', label='Target class',
orientation='horizontal', items=[], contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display, self, 'size_calc_idx', label='Size',
orientation='horizontal',
items=list(zip(*self.SIZE_CALCULATION))[0], contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(
box_display, self, 'size_log_scale',
label='Log scale factor', minValue=1, maxValue=100, ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
gui.checkBox(
box_plot, self, 'tooltips_enabled', label='Enable tooltips',
callback=self.update_tooltip_enabled)
gui.checkBox(
box_plot, self, 'show_legend', label='Show legend',
callback=self.update_show_legend)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(
QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Expanding)
# MAIN AREA
# The QGraphicsScene doesn't actually require a parent, but not linking
# the widget to the scene causes errors and a segfault on close due to
# the way Qt deallocates memory and deletes objects.
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QtGui.QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer()
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
class OWPythagorasTree(OWWidget):
name = 'Pythagorean Tree'
description = 'Pythagorean Tree visualization for tree like-structures.'
icon = 'icons/PythagoreanTree.svg'
priority = 1000
inputs = [('Tree', Tree, 'set_tree')]
outputs = [('Selected Data', Table)]
# Enable the save as feature
graph_name = 'scene'
# Settings
depth_limit = settings.ContextSetting(10)
target_class_index = settings.ContextSetting(0)
size_calc_idx = settings.Setting(0)
size_log_scale = settings.Setting(2)
tooltips_enabled = settings.Setting(True)
show_legend = settings.Setting(False)
GENERAL, CLASSIFICATION, REGRESSION = range(3)
LEGEND_OPTIONS = {
'corner': Anchorable.BOTTOM_RIGHT,
'offset': (10, 10),
}
def __init__(self):
super().__init__()
# Instance variables
self.tree_type = self.GENERAL
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
# The +1 is there so that we don't get division by 0 exceptions
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# Color modes for regression trees
self.REGRESSION_COLOR_CALC = [
('None', lambda _, __: QtGui.QColor(255, 255, 255)),
('Class mean', self._color_class_mean),
('Standard deviation', self._color_stddev),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.info = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(
box_display, self, 'depth_limit', label='Depth', ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(
box_display, self, 'target_class_index', label='Target class',
orientation='horizontal', items=[], contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display, self, 'size_calc_idx', label='Size',
orientation='horizontal',
items=list(zip(*self.SIZE_CALCULATION))[0], contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(
box_display, self, 'size_log_scale',
label='Log scale factor', minValue=1, maxValue=100, ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
gui.checkBox(
box_plot, self, 'tooltips_enabled', label='Enable tooltips',
callback=self.update_tooltip_enabled)
gui.checkBox(
box_plot, self, 'show_legend', label='Show legend',
callback=self.update_show_legend)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(
QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Expanding)
# MAIN AREA
# The QGraphicsScene doesn't actually require a parent, but not linking
# the widget to the scene causes errors and a segfault on close due to
# the way Qt deallocates memory and deletes objects.
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QtGui.QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer()
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
def set_tree(self, model=None):
"""When a different tree is given."""
self.clear()
self.model = model
if model is not None:
# We need to know what kind of tree we have in order to properly
# show colors and tooltips
if isinstance(model, TreeClassifier):
self.tree_type = self.CLASSIFICATION
elif isinstance(model, TreeRegressor):
self.tree_type = self.REGRESSION
else:
self.tree_type = self.GENERAL
self.instances = model.instances
# this bit is important for the regression classifier
if self.instances is not None and \
self.instances.domain != model.domain:
self.clf_dataset = Table.from_table(
self.model.domain, self.instances)
else:
self.clf_dataset = self.instances
self.tree_adapter = self._get_tree_adapter(self.model)
self.color_palette = self._tree_specific('_get_color_palette')()
self.ptree.clear()
self.ptree.set_tree(self.tree_adapter)
self.ptree.set_tooltip_func(self._tree_specific('_get_tooltip'))
self.ptree.set_node_color_func(
self._tree_specific('_get_node_color')
)
self._tree_specific('_update_legend_colors')()
self._update_legend_visibility()
self._update_info_box()
self._update_depth_slider()
self._tree_specific('_update_target_class_combo')()
self._update_main_area()
# Get meta variables describing pythagoras tree if given from
# forest.
if hasattr(model, 'meta_size_calc_idx'):
self.size_calc_idx = model.meta_size_calc_idx
if hasattr(model, 'meta_size_log_scale'):
self.size_log_scale = model.meta_size_log_scale
# Updating the size calc redraws the whole tree
if hasattr(model, 'meta_size_calc_idx') or \
hasattr(model, 'meta_size_log_scale'):
self.update_size_calc()
# The target class can also be passed from the meta properties
if hasattr(model, 'meta_target_class_index'):
self.target_class_index = model.meta_target_class_index
self.update_colors()
# TODO this messes up the viewport in pythagoras tree viewer
# it seems the viewport doesn't reset its size if this is applied
# if hasattr(model, 'meta_depth_limit'):
# self.depth_limit = model.meta_depth_limit
# self.update_depth()
def clear(self):
"""Clear all relevant data from the widget."""
self.model = None
self.instances = None
self.clf_dataset = None
self.tree_adapter = None
if self.legend is not None:
self.scene.removeItem(self.legend)
self.legend = None
self.ptree.clear()
self._clear_info_box()
self._clear_target_class_combo()
self._clear_depth_slider()
self._update_log_scale_slider()
# CONTROL AREA CALLBACKS
def update_depth(self):
"""This method should be called when the depth changes"""
self.ptree.set_depth_limit(self.depth_limit)
def update_colors(self):
"""When the target class / node coloring needs to be updated."""
self.ptree.target_class_has_changed()
self._tree_specific('_update_legend_colors')()
def update_size_calc(self):
"""When the tree size calculation is updated."""
self._update_log_scale_slider()
self.invalidate_tree()
def invalidate_tree(self):
"""When the tree needs to be recalculated. E.g. change of size calc."""
if self.model is not None:
self.tree_adapter = self._get_tree_adapter(self.model)
self.ptree.set_tree(self.tree_adapter)
self.ptree.set_depth_limit(self.depth_limit)
self._update_main_area()
def update_tooltip_enabled(self):
"""When the tooltip visibility is changed and need to be updated."""
if self.tooltips_enabled:
self.ptree.set_tooltip_func(
self._tree_specific('_get_tooltip')
)
else:
self.ptree.set_tooltip_func(lambda _: None)
self.ptree.tooltip_has_changed()
def update_show_legend(self):
"""When the legend visibility needs to be updated."""
self._update_legend_visibility()
# MODEL CHANGED CONTROL ELEMENTS UPDATE METHODS
def _update_info_box(self):
self.info.setText('Nodes: {}\nDepth: {}'.format(
self.tree_adapter.num_nodes,
self.tree_adapter.max_depth
))
def _update_depth_slider(self):
self.depth_slider.parent().setEnabled(True)
self.depth_slider.setMaximum(self.tree_adapter.max_depth)
self._set_max_depth()
def _update_legend_visibility(self):
if self.legend is not None:
self.legend.setVisible(self.show_legend)
def _update_log_scale_slider(self):
"""On calc method combo box changed."""
self.log_scale_box.parent().setEnabled(
self.SIZE_CALCULATION[self.size_calc_idx][0] == 'Logarithmic')
# MODEL REMOVED CONTROL ELEMENTS CLEAR METHODS
def _clear_info_box(self):
self.info.setText('No tree on input')
def _clear_depth_slider(self):
self.depth_slider.parent().setEnabled(False)
self.depth_slider.setMaximum(0)
def _clear_target_class_combo(self):
self.target_class_combo.clear()
self.target_class_index = 0
self.target_class_combo.setCurrentIndex(self.target_class_index)
# HELPFUL METHODS
def _set_max_depth(self):
"""Set the depth to the max depth and update appropriate actors."""
self.depth_limit = self.tree_adapter.max_depth
self.depth_slider.setValue(self.depth_limit)
def _update_main_area(self):
# refresh the scene rect, cuts away the excess whitespace, and adds
# padding for panning.
self.scene.setSceneRect(self.view.central_widget_rect())
# reset the zoom level
self.view.recalculate_and_fit()
self.view.update_anchored_items()
def _get_tree_adapter(self, model):
return SklTreeAdapter(
model.tree,
model.domain,
adjust_weight=self.SIZE_CALCULATION[self.size_calc_idx][1],
)
def onDeleteWidget(self):
"""When deleting the widget."""
super().onDeleteWidget()
self.clear()
def commit(self):
"""Commit the selected data to output."""
if self.instances is None:
self.send('Selected Data', None)
return
# this is taken almost directly from the owclassificationtreegraph.py
items = filter(lambda x: isinstance(x, SquareGraphicsItem),
self.scene.selectedItems())
data = self.tree_adapter.get_instances_in_nodes(
self.clf_dataset, [item.tree_node for item in items])
self.send('Selected Data', data)
def send_report(self):
"""Send report."""
self.report_plot()
def _tree_specific(self, method):
"""A best effort method getter that somewhat separates logic specific
to classification and regression trees.
This relies on conventional naming of specific methods, e.g.
a method name _get_tooltip would need to be defined like so:
_classification_get_tooltip and _regression_get_tooltip, since they are
both specific.
Parameters
----------
method : str
Method name that we would like to call.
Returns
-------
callable or None
"""
if self.tree_type == self.GENERAL:
return getattr(self, '_general' + method)
elif self.tree_type == self.CLASSIFICATION:
return getattr(self, '_classification' + method)
elif self.tree_type == self.REGRESSION:
return getattr(self, '_regression' + method)
else:
return None
# CLASSIFICATION TREE SPECIFIC METHODS
def _classification_update_target_class_combo(self):
self._clear_target_class_combo()
list(filter(
lambda x: isinstance(x, QtGui.QLabel),
self.target_class_combo.parent().children()
))[0].setText('Target class')
self.target_class_combo.addItem('None')
values = [c.title() for c in
self.tree_adapter.domain.class_vars[0].values]
self.target_class_combo.addItems(values)
def _classification_update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
if self.target_class_index == 0:
self.legend = OWDiscreteLegend(domain=self.model.domain,
**self.LEGEND_OPTIONS)
else:
items = (
(self.target_class_combo.itemText(self.target_class_index),
self.color_palette[self.target_class_index - 1]),
('other', QtGui.QColor('#ffffff'))
)
self.legend = OWDiscreteLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _classification_get_color_palette(self):
return [QtGui.QColor(*c) for c in self.model.domain.class_var.colors]
def _classification_get_node_color(self, adapter, tree_node):
# this is taken almost directly from the existing classification tree
# viewer
colors = self.color_palette
distribution = adapter.get_distribution(tree_node.label)[0]
total = np.sum(distribution)
if self.target_class_index:
p = distribution[self.target_class_index - 1] / total
color = colors[self.target_class_index - 1].light(200 - 100 * p)
else:
modus = np.argmax(distribution)
p = distribution[modus] / (total or 1)
color = colors[int(modus)].light(400 - 300 * p)
return color
def _classification_get_tooltip(self, node):
distribution = self.tree_adapter.get_distribution(node.label)[0]
total = int(np.sum(distribution))
if self.target_class_index:
samples = distribution[self.target_class_index - 1]
text = ''
else:
modus = np.argmax(distribution)
samples = distribution[modus]
text = self.tree_adapter.domain.class_vars[0].values[modus] + \
'<br>'
ratio = samples / np.sum(distribution)
rules = self.tree_adapter.rules(node.label)
sorted_rules = sorted(rules[:-1], key=lambda rule: rule.attr_name)
rules_str = ''
if len(rules):
rules_str += '<br>'.join(str(rule) for rule in sorted_rules)
rules_str += '<br><b>%s</b>' % rules[-1]
splitting_attr = self.tree_adapter.attribute(node.label)
return '<p>' \
+ text \
+ '{}/{} samples ({:2.3f}%)'.format(
int(samples), total, ratio * 100) \
+ '<hr>' \
+ ('Split by ' + splitting_attr.name
if not self.tree_adapter.is_leaf(node.label) else '') \
+ ('<br><br>'
if len(rules) and not self.tree_adapter.is_leaf(node.label)
else '') \
+ rules_str \
+ '</p>'
# REGRESSION TREE SPECIFIC METHODS
def _regression_update_target_class_combo(self):
self._clear_target_class_combo()
list(filter(
lambda x: isinstance(x, QtGui.QLabel),
self.target_class_combo.parent().children()
))[0].setText('Node color')
self.target_class_combo.addItems(
list(zip(*self.REGRESSION_COLOR_CALC))[0])
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _regression_update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
def _get_colors_domain(domain):
class_var = domain.class_var
start, end, pass_through_black = class_var.colors
if pass_through_black:
lst_colors = [QtGui.QColor(*c) for c
in [start, (0, 0, 0), end]]
else:
lst_colors = [QtGui.QColor(*c) for c in [start, end]]
return lst_colors
# Currently, the first index just draws the outline without any color
if self.target_class_index == 0:
self.legend = None
return
# The colors are the class mean
elif self.target_class_index == 1:
values = (np.min(self.clf_dataset.Y), np.max(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
self.legend = OWContinuousLegend(items=list(zip(values, colors)),
**self.LEGEND_OPTIONS)
# Colors are the stddev
elif self.target_class_index == 2:
values = (0, np.std(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
self.legend = OWContinuousLegend(items=list(zip(values, colors)),
**self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _regression_get_color_palette(self):
return ContinuousPaletteGenerator(
*self.tree_adapter.domain.class_var.colors)
def _regression_get_node_color(self, adapter, tree_node):
return self.REGRESSION_COLOR_CALC[self.target_class_index][1](
adapter, tree_node
)
def _color_class_mean(self, adapter, tree_node):
# calculate node colors relative to the mean of the node samples
min_mean = np.min(self.clf_dataset.Y)
max_mean = np.max(self.clf_dataset.Y)
instances = adapter.get_instances_in_nodes(self.clf_dataset, tree_node)
mean = np.mean(instances.Y)
return self.color_palette[(mean - min_mean) / (max_mean - min_mean)]
def _color_stddev(self, adapter, tree_node):
# calculate node colors relative to the standard deviation in the node
# samples
min_mean, max_mean = 0, np.std(self.clf_dataset.Y)
instances = adapter.get_instances_in_nodes(self.clf_dataset, tree_node)
std = np.std(instances.Y)
return self.color_palette[(std - min_mean) / (max_mean - min_mean)]
def _regression_get_tooltip(self, node):
total = self.tree_adapter.num_samples(
self.tree_adapter.parent(node.label))
samples = self.tree_adapter.num_samples(node.label)
ratio = samples / total
instances = self.tree_adapter.get_instances_in_nodes(
self.clf_dataset, node)
mean = np.mean(instances.Y)
std = np.std(instances.Y)
rules = self.tree_adapter.rules(node.label)
sorted_rules = sorted(rules[:-1], key=lambda rule: rule.attr_name)
rules_str = ''
if len(rules):
rules_str += '<br>'.join(str(rule) for rule in sorted_rules)
rules_str += '<br><b>%s</b>' % rules[-1]
splitting_attr = self.tree_adapter.attribute(node.label)
return '<p>Mean: {:2.3f}'.format(mean) \
+ '<br>Standard deviation: {:2.3f}'.format(std) \
+ '<br>{}/{} samples ({:2.3f}%)'.format(
int(samples), total, ratio * 100) \
+ '<hr>' \
+ ('Split by ' + splitting_attr.name
if not self.tree_adapter.is_leaf(node.label) else '') \
+ ('<br><br>' if len(rules) and not self.tree_adapter.is_leaf(
node.label) else '') \
+ rules_str \
+ '</p>'
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.infolabel = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(
box_display, self, 'depth_limit', label='Depth', ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(
box_display, self, 'target_class_index', label='Target class',
orientation=Qt.Horizontal, items=[], contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display, self, 'size_calc_idx', label='Size',
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[0], contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(
box_display, self, 'size_log_scale',
label='Log scale factor', minValue=1, maxValue=100, ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
self.cb_show_tooltips = gui.checkBox(
box_plot, self, 'tooltips_enabled', label='Enable tooltips',
callback=self.update_tooltip_enabled)
self.cb_show_legend = gui.checkBox(
box_plot, self, 'show_legend', label='Show legend',
callback=self.update_show_legend)
gui.button(self.controlArea, self, label="Redraw", callback=self.redraw)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(
QSizePolicy.Preferred, QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
class OWPythagorasTree(OWWidget):
name = 'Pythagorean Tree'
description = 'Pythagorean Tree visualization for tree like-structures.'
icon = 'icons/PythagoreanTree.svg'
keywords = ["fractal"]
priority = 1000
class Inputs:
tree = Input("Tree", TreeModel)
class Outputs:
selected_data = Output("Selected Data", Table, default=True)
annotated_data = Output(ANNOTATED_DATA_SIGNAL_NAME, Table)
# Enable the save as feature
graph_name = 'scene'
# Settings
depth_limit = settings.ContextSetting(10)
target_class_index = settings.ContextSetting(0)
size_calc_idx = settings.Setting(0)
size_log_scale = settings.Setting(2)
tooltips_enabled = settings.Setting(True)
show_legend = settings.Setting(False)
LEGEND_OPTIONS = {
'corner': Anchorable.BOTTOM_RIGHT,
'offset': (10, 10),
}
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.instances = None
self.clf_dataset = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
('Normal', lambda x: x),
('Square root', lambda x: sqrt(x)),
('Logarithmic', lambda x: log(x * self.size_log_scale + 1)),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, 'Tree Info')
self.infolabel = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, 'Display Settings')
self.depth_slider = gui.hSlider(
box_display, self, 'depth_limit', label='Depth', ticks=False,
callback=self.update_depth)
self.target_class_combo = gui.comboBox(
box_display, self, 'target_class_index', label='Target class',
orientation=Qt.Horizontal, items=[], contentsLength=8,
callback=self.update_colors)
self.size_calc_combo = gui.comboBox(
box_display, self, 'size_calc_idx', label='Size',
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[0], contentsLength=8,
callback=self.update_size_calc)
self.log_scale_box = gui.hSlider(
box_display, self, 'size_log_scale',
label='Log scale factor', minValue=1, maxValue=100, ticks=False,
callback=self.invalidate_tree)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, 'Plot Properties')
self.cb_show_tooltips = gui.checkBox(
box_plot, self, 'tooltips_enabled', label='Enable tooltips',
callback=self.update_tooltip_enabled)
self.cb_show_legend = gui.checkBox(
box_plot, self, 'show_legend', label='Show legend',
callback=self.update_show_legend)
gui.button(self.controlArea, self, label="Redraw", callback=self.redraw)
# Stretch to fit the rest of the unsused area
gui.rubber(self.controlArea)
self.controlArea.setSizePolicy(
QSizePolicy.Preferred, QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
@Inputs.tree
def set_tree(self, model=None):
"""When a different tree is given."""
self.clear()
self.model = model
if model is not None:
self.instances = model.instances
# this bit is important for the regression classifier
if self.instances is not None and \
self.instances.domain != model.domain:
self.clf_dataset = self.instances.transform(self.model.domain)
else:
self.clf_dataset = self.instances
self.tree_adapter = self._get_tree_adapter(self.model)
self.ptree.clear()
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self._update_depth_slider()
self.color_palette = self.ptree.root.color_palette
self._update_legend_colors()
self._update_legend_visibility()
self._update_info_box()
self._update_target_class_combo()
self._update_main_area()
# The target class can also be passed from the meta properties
# This must be set after `_update_target_class_combo`
if hasattr(model, 'meta_target_class_index'):
self.target_class_index = model.meta_target_class_index
self.update_colors()
# Get meta variables describing what the settings should look like
# if the tree is passed from the Pythagorean forest widget.
if hasattr(model, 'meta_size_calc_idx'):
self.size_calc_idx = model.meta_size_calc_idx
self.update_size_calc()
# TODO There is still something wrong with this
# if hasattr(model, 'meta_depth_limit'):
# self.depth_limit = model.meta_depth_limit
# self.update_depth()
self.Outputs.annotated_data.send(create_annotated_table(self.instances, None))
def clear(self):
"""Clear all relevant data from the widget."""
self.model = None
self.instances = None
self.clf_dataset = None
self.tree_adapter = None
if self.legend is not None:
self.scene.removeItem(self.legend)
self.legend = None
self.ptree.clear()
self._clear_info_box()
self._clear_target_class_combo()
self._clear_depth_slider()
self._update_log_scale_slider()
def update_depth(self):
"""This method should be called when the depth changes"""
self.ptree.set_depth_limit(self.depth_limit)
def update_colors(self):
"""When the target class / node coloring needs to be updated."""
self.ptree.target_class_changed(self.target_class_index)
self._update_legend_colors()
def update_size_calc(self):
"""When the tree size calculation is updated."""
self._update_log_scale_slider()
self.invalidate_tree()
def redraw(self):
self.tree_adapter.shuffle_children()
self.invalidate_tree()
def invalidate_tree(self):
"""When the tree needs to be completely recalculated."""
if self.model is not None:
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self.ptree.set_depth_limit(self.depth_limit)
self._update_main_area()
def update_tooltip_enabled(self):
"""When the tooltip visibility is changed and need to be updated."""
self.ptree.tooltip_changed(self.tooltips_enabled)
def update_show_legend(self):
"""When the legend visibility needs to be updated."""
self._update_legend_visibility()
def _update_info_box(self):
self.infolabel.setText('Nodes: {}\nDepth: {}'.format(
self.tree_adapter.num_nodes,
self.tree_adapter.max_depth
))
def _update_depth_slider(self):
self.depth_slider.parent().setEnabled(True)
self.depth_slider.setMaximum(self.tree_adapter.max_depth)
self._set_max_depth()
def _update_legend_visibility(self):
if self.legend is not None:
self.legend.setVisible(self.show_legend)
def _update_log_scale_slider(self):
"""On calc method combo box changed."""
self.log_scale_box.parent().setEnabled(
self.SIZE_CALCULATION[self.size_calc_idx][0] == 'Logarithmic')
def _clear_info_box(self):
self.infolabel.setText('No tree on input')
def _clear_depth_slider(self):
self.depth_slider.parent().setEnabled(False)
self.depth_slider.setMaximum(0)
def _clear_target_class_combo(self):
self.target_class_combo.clear()
self.target_class_index = 0
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _set_max_depth(self):
"""Set the depth to the max depth and update appropriate actors."""
self.depth_limit = self.tree_adapter.max_depth
self.depth_slider.setValue(self.depth_limit)
def _update_main_area(self):
# refresh the scene rect, cuts away the excess whitespace, and adds
# padding for panning.
self.scene.setSceneRect(self.view.central_widget_rect())
# reset the zoom level
self.view.recalculate_and_fit()
self.view.update_anchored_items()
def _get_tree_adapter(self, model):
if isinstance(model, SklModel):
return SklTreeAdapter(model)
return TreeAdapter(model)
def onDeleteWidget(self):
"""When deleting the widget."""
super().onDeleteWidget()
self.clear()
def commit(self):
"""Commit the selected data to output."""
if self.instances is None:
self.Outputs.selected_data.send(None)
self.Outputs.annotated_data.send(None)
return
nodes = [i.tree_node.label for i in self.scene.selectedItems()
if isinstance(i, SquareGraphicsItem)]
data = self.tree_adapter.get_instances_in_nodes(nodes)
self.Outputs.selected_data.send(data)
selected_indices = self.tree_adapter.get_indices(nodes)
self.Outputs.annotated_data.send(create_annotated_table(self.instances, selected_indices))
def send_report(self):
"""Send report."""
self.report_plot()
def _update_target_class_combo(self):
self._clear_target_class_combo()
label = [x for x in self.target_class_combo.parent().children()
if isinstance(x, QLabel)][0]
if self.instances.domain.has_discrete_class:
label_text = 'Target class'
values = [c.title() for c in self.instances.domain.class_vars[0].values]
values.insert(0, 'None')
else:
label_text = 'Node color'
values = list(ContinuousTreeNode.COLOR_METHODS.keys())
label.setText(label_text)
self.target_class_combo.addItems(values)
self.target_class_combo.setCurrentIndex(self.target_class_index)
def _update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
if self.instances.domain.has_discrete_class:
self._classification_update_legend_colors()
else:
self._regression_update_legend_colors()
def _classification_update_legend_colors(self):
if self.target_class_index == 0:
self.legend = OWDiscreteLegend(domain=self.model.domain,
**self.LEGEND_OPTIONS)
else:
items = (
(self.target_class_combo.itemText(self.target_class_index),
self.color_palette[self.target_class_index - 1]),
('other', QColor('#ffffff'))
)
self.legend = OWDiscreteLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _regression_update_legend_colors(self):
def _get_colors_domain(domain):
class_var = domain.class_var
start, end, pass_through_black = class_var.colors
if pass_through_black:
lst_colors = [QColor(*c) for c
in [start, (0, 0, 0), end]]
else:
lst_colors = [QColor(*c) for c in [start, end]]
return lst_colors
# The colors are the class mean
if self.target_class_index == 1:
values = (np.min(self.clf_dataset.Y), np.max(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
items = list(zip(values, colors))
# Colors are the stddev
elif self.target_class_index == 2:
values = (0, np.std(self.clf_dataset.Y))
colors = _get_colors_domain(self.model.domain)
while len(values) != len(colors):
values.insert(1, -1)
items = list(zip(values, colors))
else:
items = None
self.legend = OWContinuousLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.data = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
("Normal", lambda x: x, "正常"),
("Square root", lambda x: sqrt(x), "平方根"),
("Logarithmic", lambda x: log(x * self.size_log_scale + 1), "对数"),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, "树信息")
self.infolabel = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, "显示设置")
# maxValue is set to a wide three-digit number to probably ensure the
# proper label width. The maximum is later set to match the tree depth
self.depth_slider = gui.hSlider(
box_display,
self,
"depth_limit",
label="深度",
ticks=False,
maxValue=900,
callback=self.update_depth,
)
self.target_class_combo = gui.comboBox(
box_display,
self,
"target_class_index",
label="目标类别",
orientation=Qt.Horizontal,
items=[],
contentsLength=8,
searchable=True,
callback=self.update_colors,
)
self.size_calc_combo = gui.comboBox(
box_display,
self,
"size_calc_idx",
label="大小",
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[2],
contentsLength=8,
callback=self.update_size_calc,
)
self.log_scale_box = gui.hSlider(
box_display,
self,
"size_log_scale",
label="对数比例因子",
minValue=1,
maxValue=100,
ticks=False,
callback=self.invalidate_tree,
)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, "绘图属性")
self.cb_show_tooltips = gui.checkBox(
box_plot,
self,
"tooltips_enabled",
label="启动工具提示",
callback=self.update_tooltip_enabled,
)
self.cb_show_legend = gui.checkBox(
box_plot,
self,
"show_legend",
label="显示图例",
callback=self.update_show_legend,
)
gui.rubber(self.controlArea)
gui.button(self.buttonsArea, self, label="重新绘制", callback=self.redraw)
self.controlArea.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
class OWPythagorasTree(OWWidget):
name = "毕达哥拉斯树(Pythagorean Tree)"
description = "类似树结构的毕达哥拉斯树可视化。"
icon = "icons/PythagoreanTree.svg"
keywords = ["fractal", "bidagelasishu", "gougushu"]
category = "可视化(Visualize)"
priority = 1000
class Inputs:
tree = Input("树(Tree)", TreeModel, replaces=["Tree"])
class Outputs:
selected_data = Output("选定的数据(Selected Data)",
Table,
default=True,
replaces=["Selected Data"])
annotated_data = Output("数据(Data)", Table, replaces=["Data"])
# Enable the save as feature
graph_name = "scene"
# Settings
settingsHandler = settings.ClassValuesContextHandler()
depth_limit = settings.ContextSetting(10)
target_class_index = settings.ContextSetting(0)
size_calc_idx = settings.Setting(0)
size_log_scale = settings.Setting(2)
tooltips_enabled = settings.Setting(True)
show_legend = settings.Setting(False)
LEGEND_OPTIONS = {
"corner": Anchorable.BOTTOM_RIGHT,
"offset": (10, 10),
}
def __init__(self):
super().__init__()
# Instance variables
self.model = None
self.data = None
# The tree adapter instance which is passed from the outside
self.tree_adapter = None
self.legend = None
self.color_palette = None
# Different methods to calculate the size of squares
self.SIZE_CALCULATION = [
("Normal", lambda x: x, "正常"),
("Square root", lambda x: sqrt(x), "平方根"),
("Logarithmic", lambda x: log(x * self.size_log_scale + 1), "对数"),
]
# CONTROL AREA
# Tree info area
box_info = gui.widgetBox(self.controlArea, "树信息")
self.infolabel = gui.widgetLabel(box_info)
# Display settings area
box_display = gui.widgetBox(self.controlArea, "显示设置")
# maxValue is set to a wide three-digit number to probably ensure the
# proper label width. The maximum is later set to match the tree depth
self.depth_slider = gui.hSlider(
box_display,
self,
"depth_limit",
label="深度",
ticks=False,
maxValue=900,
callback=self.update_depth,
)
self.target_class_combo = gui.comboBox(
box_display,
self,
"target_class_index",
label="目标类别",
orientation=Qt.Horizontal,
items=[],
contentsLength=8,
searchable=True,
callback=self.update_colors,
)
self.size_calc_combo = gui.comboBox(
box_display,
self,
"size_calc_idx",
label="大小",
orientation=Qt.Horizontal,
items=list(zip(*self.SIZE_CALCULATION))[2],
contentsLength=8,
callback=self.update_size_calc,
)
self.log_scale_box = gui.hSlider(
box_display,
self,
"size_log_scale",
label="对数比例因子",
minValue=1,
maxValue=100,
ticks=False,
callback=self.invalidate_tree,
)
# Plot properties area
box_plot = gui.widgetBox(self.controlArea, "绘图属性")
self.cb_show_tooltips = gui.checkBox(
box_plot,
self,
"tooltips_enabled",
label="启动工具提示",
callback=self.update_tooltip_enabled,
)
self.cb_show_legend = gui.checkBox(
box_plot,
self,
"show_legend",
label="显示图例",
callback=self.update_show_legend,
)
gui.rubber(self.controlArea)
gui.button(self.buttonsArea, self, label="重新绘制", callback=self.redraw)
self.controlArea.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.Expanding)
# MAIN AREA
self.scene = TreeGraphicsScene(self)
self.scene.selectionChanged.connect(self.commit)
self.view = TreeGraphicsView(self.scene, padding=(150, 150))
self.view.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.view)
self.ptree = PythagorasTreeViewer(self)
self.scene.addItem(self.ptree)
self.view.set_central_widget(self.ptree)
self.resize(800, 500)
# Clear the widget to correctly set the intial values
self.clear()
@Inputs.tree
def set_tree(self, model=None):
"""When a different tree is given."""
self.closeContext()
self.clear()
self.model = model
if model is not None:
self.data = model.instances
self._update_target_class_combo()
self.tree_adapter = self._get_tree_adapter(self.model)
self.ptree.clear()
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self._update_depth_slider()
self.color_palette = self.ptree.root.color_palette
self._update_legend_colors()
self._update_legend_visibility()
self._update_info_box()
self._update_main_area()
self.openContext(
model.domain.class_var if model.domain is not None else None)
self.update_depth()
# The forest widget sets the following attributes on the tree,
# describing the settings on the forest widget. To keep the tree
# looking the same as on the forest widget, we prefer these settings to
# context settings, if set.
if hasattr(model, "meta_target_class_index"):
self.target_class_index = model.meta_target_class_index
self.update_colors()
if hasattr(model, "meta_size_calc_idx"):
self.size_calc_idx = model.meta_size_calc_idx
self.update_size_calc()
if hasattr(model, "meta_depth_limit"):
self.depth_limit = model.meta_depth_limit
self.update_depth()
self.Outputs.annotated_data.send(
create_annotated_table(self.data, None))
def clear(self):
"""Clear all relevant data from the widget."""
self.model = None
self.data = None
self.tree_adapter = None
if self.legend is not None:
self.scene.removeItem(self.legend)
self.legend = None
self.ptree.clear()
self._clear_info_box()
self._clear_target_class_combo()
self._clear_depth_slider()
self._update_log_scale_slider()
def update_depth(self):
"""This method should be called when the depth changes"""
self.ptree.set_depth_limit(self.depth_limit)
def update_colors(self):
"""When the target class / node coloring needs to be updated."""
self.ptree.target_class_changed(self.target_class_index)
self._update_legend_colors()
def update_size_calc(self):
"""When the tree size calculation is updated."""
self._update_log_scale_slider()
self.invalidate_tree()
def redraw(self):
if self.data is None:
return
self.tree_adapter.shuffle_children()
self.invalidate_tree()
def invalidate_tree(self):
"""When the tree needs to be completely recalculated."""
if self.model is not None:
self.ptree.set_tree(
self.tree_adapter,
weight_adjustment=self.SIZE_CALCULATION[self.size_calc_idx][1],
target_class_index=self.target_class_index,
)
self.ptree.set_depth_limit(self.depth_limit)
self._update_main_area()
def update_tooltip_enabled(self):
"""When the tooltip visibility is changed and need to be updated."""
self.ptree.tooltip_changed(self.tooltips_enabled)
def update_show_legend(self):
"""When the legend visibility needs to be updated."""
self._update_legend_visibility()
def _update_info_box(self):
self.infolabel.setText("节点: {}\n深度: {}".format(
self.tree_adapter.num_nodes, self.tree_adapter.max_depth))
def _update_depth_slider(self):
self.depth_slider.parent().setEnabled(True)
self.depth_slider.setMaximum(self.tree_adapter.max_depth)
self._set_max_depth()
def _update_legend_visibility(self):
if self.legend is not None:
self.legend.setVisible(self.show_legend)
def _update_log_scale_slider(self):
"""On calc method combo box changed."""
self.log_scale_box.parent().setEnabled(
self.SIZE_CALCULATION[self.size_calc_idx][0] == "Logarithmic")
def _clear_info_box(self):
self.infolabel.setText("没有树输入")
def _clear_depth_slider(self):
self.depth_slider.parent().setEnabled(False)
self.depth_slider.setMaximum(0)
def _clear_target_class_combo(self):
self.target_class_combo.clear()
self.target_class_index = -1
def _set_max_depth(self):
"""Set the depth to the max depth and update appropriate actors."""
self.depth_limit = self.tree_adapter.max_depth
self.depth_slider.setValue(self.depth_limit)
def _update_main_area(self):
# refresh the scene rect, cuts away the excess whitespace, and adds
# padding for panning.
self.scene.setSceneRect(self.view.central_widget_rect())
# reset the zoom level
self.view.recalculate_and_fit()
self.view.update_anchored_items()
def _get_tree_adapter(self, model):
if isinstance(model, SklModel):
return SklTreeAdapter(model)
return TreeAdapter(model)
def onDeleteWidget(self):
"""When deleting the widget."""
super().onDeleteWidget()
self.clear()
def commit(self):
"""Commit the selected data to output."""
if self.data is None:
self.Outputs.selected_data.send(None)
self.Outputs.annotated_data.send(None)
return
nodes = [
i.tree_node.label for i in self.scene.selectedItems()
if isinstance(i, SquareGraphicsItem)
]
data = self.tree_adapter.get_instances_in_nodes(nodes)
self.Outputs.selected_data.send(data)
selected_indices = self.tree_adapter.get_indices(nodes)
self.Outputs.annotated_data.send(
create_annotated_table(self.data, selected_indices))
def send_report(self):
"""Send report."""
self.report_plot()
def _update_target_class_combo(self):
self._clear_target_class_combo()
label = [
x for x in self.target_class_combo.parent().children()
if isinstance(x, QLabel)
][0]
if self.data.domain.has_discrete_class:
label_text = "目标类别"
values = [c.title() for c in self.data.domain.class_vars[0].values]
values.insert(0, "None")
else:
label_text = "Node color"
values = list(ContinuousTreeNode.COLOR_METHODS.keys())
label.setText(label_text)
self.target_class_combo.addItems(values)
# set it to 0, context will change if required
self.target_class_index = 0
def _update_legend_colors(self):
if self.legend is not None:
self.scene.removeItem(self.legend)
if self.data.domain.has_discrete_class:
self._classification_update_legend_colors()
else:
self._regression_update_legend_colors()
def _classification_update_legend_colors(self):
if self.target_class_index == 0:
self.legend = OWDiscreteLegend(domain=self.model.domain,
**self.LEGEND_OPTIONS)
else:
items = (
(
self.target_class_combo.itemText(self.target_class_index),
self.color_palette[self.target_class_index - 1],
),
("other", QColor("#ffffff")),
)
self.legend = OWDiscreteLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
def _regression_update_legend_colors(self):
# The colors are the class mean
palette = self.model.domain.class_var.palette
if self.target_class_index == 1:
items = ((np.min(self.data.Y), np.max(self.data.Y)), palette)
# Colors are the stddev
elif self.target_class_index == 2:
items = ((0, np.std(self.data.Y)), palette)
else:
items = None
self.legend = OWContinuousLegend(items=items, **self.LEGEND_OPTIONS)
self.legend.setVisible(self.show_legend)
self.scene.addItem(self.legend)
