Source code for gui.core.regressionMethods.ARD

from PyQt5 import QtWidgets
from sklearn.linear_model import ARDRegression

from gui.ui.ARD import Ui_Form
from gui.util.Modules import Modules



[docs]
class Ui_Form(Ui_Form, ARDRegression, Modules):

[docs]
    def setupUi(self, Form):
        super().setupUi(Form)
        self.checkMinAndMax()
        self.connectWidgets()



[docs]
    def get_widget(self):
        return self.formGroupBox



[docs]
    def setHidden(self, bool):
        self.get_widget().setHidden(bool)



[docs]
    def toggle_params(self, widgets, state):
        for w in widgets:
            if state:
                w.setHidden(True)
            else:
                w.setHidden(False)



[docs]
    def connectWidgets(self):
        alphawidgets = [
            self.alpha1Label,
            self.alpha1DoubleSpinBox,
            self.alpha2Label,
            self.alpha2DoubleSpinBox,
        ]
        lambdawidgets = [
            self.lambdaLabel,
            self.lambdaDoubleSpinBox,
            self.lambdaLabel_2,
            self.lambdaDoubleSpinBox_2,
        ]
        self.alpha_checkbox.stateChanged.connect(
            lambda: self.toggle_params(alphawidgets, self.alpha_checkbox.isChecked())
        )
        self.lambda_checkbox.stateChanged.connect(
            lambda: self.toggle_params(lambdawidgets, self.lambda_checkbox.isChecked())
        )
        self.alpha_checkbox.setChecked(True)
        self.lambda_checkbox.setChecked(True)



[docs]
    def run(self):
        params = {
            "n_iter": self.numOfIterationsSpinBox.value(),
            "tol": self.toleranceDoubleSpinBox.value(),
            "alpha_1": self.alpha1DoubleSpinBox.value(),
            "alpha_2": self.alpha2DoubleSpinBox.value(),
            "lambda_1": self.lambdaDoubleSpinBox.value(),
            "lambda_2": self.lambdaDoubleSpinBox_2.value(),
            "threshold_lambda": self.thresholdLambdaSpinBox.value(),
            "fit_intercept": self.fitInterceptCheckBox.isChecked(),
            "normalize": self.normalizeCheckBox.isChecked(),
        }
        return params, self.getChangedValues(params, ARDRegression())




if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    Form = QtWidgets.QWidget()
    ui = Ui_Form()
    ui.setupUi(Form)
    Form.show()
    sys.exit(app.exec_())