Source code for gui.core.regressionMethods.RF

from PyQt5 import QtWidgets
from sklearn.ensemble import RandomForestRegressor

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



[docs]
class Ui_Form(Ui_Form, RandomForestRegressor, 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 connectWidgets(self):
        pass



[docs]
    def run(self):
        params = {
            "n_estimators": int(self.numEstSpinBox.value()),
            "max_depth": int(self.max_depthSpinBox.value()),
            "random_state": 0,
        }
        return params, self.getChangedValues(params, RandomForestRegressor())




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_())