plots

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)