From d3e758bf86a574b06b59c7253dff94a6d0a1b6d8 Mon Sep 17 00:00:00 2001 From: Michael <michael.mutote@stud.th-deg.de> Date: Sat, 23 Dec 2023 08:30:57 -0800 Subject: [PATCH] Merge remote-tracking branch 'origin/main' # Conflicts: DisplayWindow.py --- DisplayWindow.py | 22 +++++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/DisplayWindow.py b/DisplayWindow.py index 0bb95e8..c4f3636 100644 --- a/DisplayWindow.py +++ b/DisplayWindow.py @@ -64,7 +64,7 @@ class MainWindow(QMainWindow): # Create a plot widget graphWidget = pg.PlotWidget(self) graphWidget.setBackground('w') - graphWidget.plot([1, 2, 3], [6, 8, 2]) + graphWidget.plot(RegressionModel.diamonds['carat'], RegressionModel.diamonds['price']) # Add the plot widget to the grid layout grid_layout.addWidget(self.calculate_button, 2, 0, 1, 2) @@ -100,14 +100,30 @@ class Advanced(QMainWindow): # Create a grid layout and set it to the central widget grid_layout = QGridLayout(central_widget) + self.graph_selector_X = QComboBox(self) + self.graph_selector_Y = QComboBox(self) + self.regression_model = QComboBox(self) 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.checkboxes = [] - grid_layout.addWidget(QLabel("select features to include in Modelling"), 0, 0) + + + grid_layout.addWidget(QLabel("select features to include in Modelling"), 0, 0, 1, 0) + # grid_layout.addWidget(self.graph_selector_X, 1, 2) + # grid_layout.addWidget(self.graph_selector_Y, 1, 3) + grid_layout.addWidget(self.regression_model, 2, 1) + # grid_layout.addWidget(QLabel("X-PLOT", self), 0, 2) + # grid_layout.addWidget(QLabel("Y-PLOT", self), 0, 3) + grid_layout.addWidget(QLabel("Regression Model : ", self), 2, 0,1,1) + + for i, label in enumerate(self.check_labels): checkbox = QCheckBox(label, self) checkbox.setChecked(True) checkbox.stateChanged.connect(self.handle_checkbox_state) # Connect signal - grid_layout.addWidget(checkbox, i, 1) + grid_layout.addWidget(checkbox, 1, i) self.checkboxes.append(checkbox) # Store checkboxes in the list def handle_checkbox_state(self): -- GitLab