Skip to content
Snippets Groups Projects
Commit bff88c7b authored by Michael Mutote's avatar Michael Mutote
Browse files

plots

parent e11011ec
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment