def __init__(self):
super().__init__()
#: input data
self.data = None
self.class_var = None
#: Current variable discretization state
self.var_state = {}
#: Saved variable discretization settings (context setting)
self.saved_var_states = {}
self.method = Methods.Default
self.k = 5
self.cutpoints = ()
box = gui.vBox(self.controlArea, self.tr("Default Discretization"))
self._default_method_ = 0
self.default_bbox = rbox = gui.radioButtons(
box, self, "_default_method_", callback=self._default_disc_changed)
self.default_button_group = bg = rbox.findChild(QButtonGroup)
bg.buttonClicked[int].connect(self.set_default_method)
rb = gui.hBox(rbox)
self.left = gui.vBox(rb)
right = gui.vBox(rb)
rb.layout().setStretch(0, 1)
rb.layout().setStretch(1, 1)
self.options = [
(Methods.Default, self.tr("Default")),
(Methods.Leave, self.tr("Leave numeric")),
(Methods.MDL, self.tr("Entropy-MDL discretization")),
(Methods.EqualFreq, self.tr("Equal-frequency discretization")),
(Methods.EqualWidth, self.tr("Equal-width discretization")),
(Methods.Remove, self.tr("Remove numeric variables")),
(Methods.Custom, self.tr("Manual")),
]
for id_, opt in self.options[1:]:
t = gui.appendRadioButton(rbox, opt)
bg.setId(t, id_)
t.setChecked(id_ == self.default_method)
[right, self.left][opt.startswith("Equal")].layout().addWidget(t)
def _intbox(parent, attr, callback):
box = gui.indentedBox(parent)
s = gui.spin(
box, self, attr, minv=2, maxv=10, label="Num. of intervals:",
callback=callback)
s.setMaximumWidth(60)
s.setAlignment(Qt.AlignRight)
gui.rubber(s.box)
return box.box
self.k_general = _intbox(self.left, "default_k",
self._default_disc_changed)
self.k_general.layout().setContentsMargins(0, 0, 0, 0)
def manual_cut_editline(text="", enabled=True) -> QLineEdit:
edit = QLineEdit(
text=text,
placeholderText="e.g. 0.0, 0.5, 1.0",
toolTip="Enter fixed discretization cut points (a comma "
"separated list of strictly increasing numbers e.g. "
"0.0, 0.5, 1.0).",
enabled=enabled,
)
@edit.textChanged.connect
def update():
validator = edit.validator()
if validator is not None:
state, _, _ = validator.validate(edit.text(), 0)
else:
state = QValidator.Acceptable
palette = edit.palette()
colors = {
QValidator.Intermediate: (Qt.yellow, Qt.black),
QValidator.Invalid: (Qt.red, Qt.black),
}.get(state, None)
if colors is None:
palette = QPalette()
else:
palette.setColor(QPalette.Base, colors[0])
palette.setColor(QPalette.Text, colors[1])
cr = edit.cursorRect()
p = edit.mapToGlobal(cr.bottomRight())
edit.setPalette(palette)
if state != QValidator.Acceptable and edit.isVisible():
show_tip(edit, p, edit.toolTip(), textFormat=Qt.RichText)
else:
show_tip(edit, p, "")
return edit
self.manual_cuts_edit = manual_cut_editline(
text=", ".join(map(str, self.default_cutpoints)),
enabled=self.default_method == Methods.Custom,
)
def set_manual_default_cuts():
text = self.manual_cuts_edit.text()
self.default_cutpoints = tuple(
float(s.strip()) for s in text.split(",") if s.strip())
self._default_disc_changed()
self.manual_cuts_edit.editingFinished.connect(set_manual_default_cuts)
validator = IncreasingNumbersListValidator()
self.manual_cuts_edit.setValidator(validator)
ibox = gui.indentedBox(right, orientation=Qt.Horizontal)
ibox.layout().addWidget(self.manual_cuts_edit)
right.layout().addStretch(10)
self.left.layout().addStretch(10)
self.connect_control(
"default_cutpoints",
lambda values: self.manual_cuts_edit.setText(", ".join(map(str, values)))
)
vlayout = QHBoxLayout()
box = gui.widgetBox(
self.controlArea, "Individual Attribute Settings",
orientation=vlayout, spacing=8
)
# List view with all attributes
self.varview = ListViewSearch(
selectionMode=QListView.ExtendedSelection,
uniformItemSizes=True,
)
self.varview.setItemDelegate(DiscDelegate())
self.varmodel = itemmodels.VariableListModel()
self.varview.setModel(self.varmodel)
self.varview.selectionModel().selectionChanged.connect(
self._var_selection_changed
)
vlayout.addWidget(self.varview)
# Controls for individual attr settings
self.bbox = controlbox = gui.radioButtons(
box, self, "method", callback=self._disc_method_changed
)
vlayout.addWidget(controlbox)
self.variable_button_group = bg = controlbox.findChild(QButtonGroup)
for id_, opt in self.options[:5]:
b = gui.appendRadioButton(controlbox, opt)
bg.setId(b, id_)
self.k_specific = _intbox(controlbox, "k", self._disc_method_changed)
gui.appendRadioButton(controlbox, "Remove attribute", id=Methods.Remove)
b = gui.appendRadioButton(controlbox, "Manual", id=Methods.Custom)
self.manual_cuts_specific = manual_cut_editline(
text=", ".join(map(str, self.cutpoints)),
enabled=self.method == Methods.Custom
)
self.manual_cuts_specific.setValidator(validator)
b.toggled[bool].connect(self.manual_cuts_specific.setEnabled)
def set_manual_cuts():
text = self.manual_cuts_specific.text()
points = [t for t in text.split(",") if t.split()]
self.cutpoints = tuple(float(t) for t in points)
self._disc_method_changed()
self.manual_cuts_specific.editingFinished.connect(set_manual_cuts)
self.connect_control(
"cutpoints",
lambda values: self.manual_cuts_specific.setText(", ".join(map(str, values)))
)
ibox = gui.indentedBox(controlbox, orientation=Qt.Horizontal)
self.copy_current_to_manual_button = b = FixedSizeButton(
text="CC", toolTip="Copy the current cut points to manual mode",
enabled=False
)
b.clicked.connect(self._copy_to_manual)
ibox.layout().addWidget(self.manual_cuts_specific)
ibox.layout().addWidget(b)
gui.rubber(controlbox)
controlbox.setEnabled(False)
bg.button(self.method)
self.controlbox = controlbox
gui.auto_apply(self.buttonsArea, self, "autosend")
self._update_spin_positions()
def _create_buttons(self, box):
"""Create radio buttons"""
def intspin():
s = QSpinBox(self)
s.setMinimum(2)
s.setMaximum(10)
s.setFixedWidth(60)
s.setAlignment(Qt.AlignRight)
s.setContentsMargins(0, 0, 0, 0)
return s, s.valueChanged
def widthline(validator):
s = QLineEdit(self)
s.setFixedWidth(60)
s.setAlignment(Qt.AlignRight)
s.setValidator(validator)
s.setContentsMargins(0, 0, 0, 0)
return s, s.textChanged
def manual_cut_editline(text="", enabled=True) -> QLineEdit:
edit = QLineEdit(
text=text,
placeholderText="e.g. 0.0, 0.5, 1.0",
toolTip='<p style="white-space:pre">' +
'Enter cut points as a comma-separate list of \n'
'strictly increasing numbers e.g. 0.0, 0.5, 1.0).</p>',
enabled=enabled,
)
edit.setValidator(IncreasingNumbersListValidator())
edit.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed)
@edit.textChanged.connect
def update():
validator = edit.validator()
if validator is not None and edit.text().strip():
state, _, _ = validator.validate(edit.text(), 0)
else:
state = QValidator.Acceptable
palette = edit.palette()
colors = {
QValidator.Intermediate: (Qt.yellow, Qt.black),
QValidator.Invalid: (Qt.red, Qt.black),
}.get(state, None)
if colors is None:
palette = QPalette()
else:
palette.setColor(QPalette.Base, colors[0])
palette.setColor(QPalette.Text, colors[1])
cr = edit.cursorRect()
p = edit.mapToGlobal(cr.bottomRight())
edit.setPalette(palette)
if state != QValidator.Acceptable and edit.isVisible():
validator.show_tip(edit, p, edit.toolTip(),
textFormat=Qt.RichText)
else:
validator.show_tip(edit, p, "")
return edit, edit.textChanged
children = []
def button(id_, *controls, stretch=True):
layout = QHBoxLayout()
desc = Options[id_]
button = QRadioButton(desc.label)
button.setToolTip(desc.tooltip)
self.button_group.addButton(button, id_)
layout.addWidget(button)
if controls:
if stretch:
layout.addStretch(1)
for c, signal in controls:
layout.addWidget(c)
if signal is not None:
@signal.connect
def arg_changed():
self.button_group.button(id_).setChecked(True)
self.update_hints(id_)
children.append(layout)
button_box.layout().addLayout(layout)
return (*controls, (None, ))[0][0]
button_box = gui.vBox(box)
button_box.layout().setSpacing(0)
button_box.setSizePolicy(QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Preferred))
self.button_group = QButtonGroup(self)
self.button_group.idClicked.connect(self.update_hints)
button(Methods.Keep)
button(Methods.Remove)
self.binning_spin = button(Methods.Binning, intspin())
validator = QDoubleValidator()
validator.setBottom(0)
self.width_line = button(Methods.FixedWidth, widthline(validator))
self.width_time_unit = u = QComboBox(self)
u.setContentsMargins(0, 0, 0, 0)
u.addItems([unit + "(s)" for unit in time_units])
validator = QIntValidator()
validator.setBottom(1)
self.width_time_line = button(Methods.FixedWidthTime,
widthline(validator),
(u, u.currentTextChanged))
self.freq_spin = button(Methods.EqualFreq, intspin())
self.width_spin = button(Methods.EqualWidth, intspin())
button(Methods.MDL)
self.copy_to_custom = FixedSizeButton(
text="CC", toolTip="Copy the current cut points to manual mode")
self.copy_to_custom.clicked.connect(self._copy_to_manual)
self.threshold_line = button(Methods.Custom,
manual_cut_editline(),
(self.copy_to_custom, None),
stretch=False)
button(Methods.Default)
maxheight = max(w.sizeHint().height() for w in children)
for w in children:
w.itemAt(0).widget().setFixedHeight(maxheight)
button_box.layout().addStretch(1)