Skip to content

Commit 8274180

Browse files
committed
make Random Forest default classifier for new projects
1 parent 3012c6d commit 8274180

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/jabs/ui/main_control_widget.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
from .k_fold_slider_widget import KFoldSliderWidget
3131
from .label_count_widget import FrameLabelCountWidget
3232

33+
DEFAULT_CLASSIFIER = ClassifierType.RANDOM_FOREST
34+
3335

3436
class MainControlWidget(QtWidgets.QWidget):
3537
"""Controls for classifier training, labeling, and settings.
@@ -157,6 +159,12 @@ def __init__(self, *args, **kwargs):
157159
for classifier, name in classifier_types.items():
158160
self._classifier_selection.addItem(name, userData=classifier)
159161

162+
# Set default classifier as the initial selection
163+
for i in range(self._classifier_selection.count()):
164+
if self._classifier_selection.itemData(i) == DEFAULT_CLASSIFIER:
165+
self._classifier_selection.setCurrentIndex(i)
166+
break
167+
160168
# slider to set number of times to train/test
161169
self._kslider = KFoldSliderWidget()
162170
self._kslider.valueChanged.connect(self.kfold_changed)

0 commit comments

Comments
 (0)