Source code for gui.core.crossValidateMethods.cv_RF
from PyQt5 import QtWidgets
from sklearn.ensemble import RandomForestRegressor
from gui.ui.cv_RF import Ui_Form
from gui.util.Modules import Modules
[docs]
class Ui_Form(Ui_Form, RandomForestRegressor, Modules):
[docs]
    def run(self):
        params = {
            'n_estimators': [int(i) for i in
                self.numEstlineEdit.text().split(',')],
            'max_depth': [int(i) for i in
                self.maxdepthlineEdit.text().split(',')],
            'random_state': [0]
        }
        return params
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_())