From bff88c7bbf3f21101a120dc4a32cde7978e862e9 Mon Sep 17 00:00:00 2001
From: Michael <michael.mutote@stud.th-deg.de>
Date: Sat, 23 Dec 2023 12:48:51 -0800
Subject: [PATCH] plots
---
DisplayWindow.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/DisplayWindow.py b/DisplayWindow.py
index c4f3636..3c251f6 100644
--- a/DisplayWindow.py
+++ b/DisplayWindow.py
@@ -106,7 +106,7 @@ class Advanced(QMainWindow):
self.check_labels = ['Cut', 'Colour', 'Clarity', 'Carat', 'Depth', 'Table', 'X', 'Y', 'Z']
self.graph_selector_X.addItems((self.check_labels + ['price']))
self.graph_selector_Y.addItems((self.check_labels + ['price']))
- self.regression_model.addItems(["Linear Regression", "Gradient Descent", "Random Forest"])
+ self.regression_model.addItems(["Linear Regression", "XG Boost", "XGBRegressor", "Neural Network"])
self.checkboxes = []
@@ -127,8 +127,11 @@ class Advanced(QMainWindow):
self.checkboxes.append(checkbox) # Store checkboxes in the list
def handle_checkbox_state(self):
+ RegressionModel.Advanced_selections = []
for i, checkbox in enumerate(self.checkboxes):
state = checkbox.checkState()
+ if state == Qt.CheckState.Unchecked:
+ RegressionModel.Advanced_selections.append(checkbox.text())
RegressionModel.Advanced_selections[checkbox.text()] = (state == Qt.CheckState.Checked)
--
GitLab