def machine_learning_fun(self):
"""This function is called when machine_learning button is clicked.
Then, this function will process the data loading.
"""
self.machine_learning = EasylearnMachineLearningRun(self.working_directory)
self.machine_learning.show()
class EasylearnMainGUI(QMainWindow, Ui_MainWindow):
"""Main GUI of the easylearn.
"""
def __init__(self):
print("Opening GUI...")
self.root_dir = os.path.dirname(eslearn.__file__)
QMainWindow.__init__(self)
Ui_MainWindow.__init__(self)
self.setupUi(self)
self.working_directory = None
self.configuration_file = ""
self.textBrowser.setText(
"Hi~, I'm easylearn. I hope I can help you finish this project successfully\n"
)
# Display start progress
self.start_process()
# Set working_directory and debug
if self.working_directory:
cgitb.enable(format="text",
display=1,
logdir=os.path.join(self.working_directory,
"log_data_loading.txt"))
else:
cgitb.enable(display=1, logdir=None)
# Connecting to functions
self.select_working_directory.triggered.connect(
self.select_workingdir_fun)
self.create_configuration_file.triggered.connect(
self.initialize_configuration_fun)
self.choose_configuration_file.triggered.connect(
self.load_configuration_fun)
self.data_loading.clicked.connect(self.data_loading_fun)
self.feature_engineering.clicked.connect(self.feature_engineering_fun)
self.machine_learning.clicked.connect(self.machine_learning_fun)
self.model_evaluation.clicked.connect(self.model_evaluation_fun)
self.run.clicked.connect(self.run_fun)
self.quit.clicked.connect(self.closeEvent_button)
# Skin
self.skins = {
"Dark": "style_Dark",
"Black": "style_black",
"DarkOrange": "style_DarkOrange",
"Gray": "style_gray",
"Blue": "style_blue",
"Navy": "style_navy",
"Classic": "style_Classic"
}
self.actionDark.triggered.connect(self.change_skin)
self.actionBlack.triggered.connect(self.change_skin)
self.actionDarkOrange.triggered.connect(self.change_skin)
self.actionGray.triggered.connect(self.change_skin)
self.actionBlue.triggered.connect(self.change_skin)
self.actionNavy.triggered.connect(self.change_skin)
self.actionClassic.triggered.connect(self.change_skin)
# Set appearance
self.set_run_appearance()
# Set initial skin
self.setStyleSheet(
pyqt5_loader.load_stylesheet_pyqt5(style="style_Dark"))
def start_process(self):
splash = QSplashScreen(
QtGui.QPixmap(os.path.join(self.root_dir, "logo/logo-upper.ico")))
splash.showMessage("... 0%",
QtCore.Qt.AlignHCenter | QtCore.Qt.AlignBottom,
QtCore.Qt.black)
splash.resize(200, 90)
splash.show()
QtWidgets.qApp.processEvents()
self.progress(splash)
splash.show()
splash.finish(self)
def progress(self, sp):
for i in range(1, 10):
time.sleep(0.2)
sp.showMessage(f"Initializing... {(i+1)*10}%",
QtCore.Qt.AlignHCenter | QtCore.Qt.AlignBottom,
QtCore.Qt.black)
QtWidgets.qApp.processEvents()
def set_run_appearance(self):
winsep = "\\"
linuxsep = "/"
self.root_dir = os.path.dirname(eslearn.__file__)
self.root_dir = self.root_dir.replace(winsep, linuxsep)
logo_upper = os.path.join(self.root_dir, "logo/logo-upper.ico")
logo_lower = os.path.join(self.root_dir, "logo/logo-lower.jpg")
logo_run = os.path.join(self.root_dir, "logo/run.png")
logo_exit = os.path.join(self.root_dir, "logo/close.png")
logo_upper = logo_upper.replace(winsep, linuxsep)
logo_lower = logo_lower.replace(winsep, linuxsep)
logo_run = logo_run.replace(winsep, linuxsep)
logo_exit = logo_exit.replace(winsep, linuxsep)
qss_logo = "#logo{" + "border-image: url(" + logo_lower + ");}" + "#logo:hover {border-radius: 0px;}"
self.logo.setStyleSheet(qss_logo)
self.setWindowTitle('easylearn')
self.setWindowIcon(QIcon(logo_upper))
# Run Icon
self.run.setIcon(QIcon(logo_run))
self.run.setIconSize(QPixmap(logo_run).size())
self.run.resize(QPixmap(logo_run).size())
# Close Icon
self.quit.setIcon(QIcon(logo_exit))
self.quit.setIconSize(QPixmap(logo_exit).size())
self.quit.resize(QPixmap(logo_exit).size())
def change_skin(self):
"""Set skins"""
sender = self.sender()
if sender:
if (sender.text() in list(self.skins.keys())):
self.setStyleSheet(
pyqt5_loader.load_stylesheet_pyqt5(
style=self.skins[sender.text()]))
else:
self.setStyleSheet(
pyqt5_loader.load_stylesheet_pyqt5(style="style_Dark"))
else:
self.setStyleSheet(
pyqt5_loader.load_stylesheet_pyqt5(style="style_Dark"))
def select_workingdir_fun(self):
"""
This function is used to select the working working_directory, then change directory to this directory.
"""
# If has selected working working_directory previously, then I set it as initial working working_directory.
if not self.working_directory:
self.working_directory = QFileDialog.getExistingDirectory(
self, "Select a working_directory", os.getcwd())
self.textBrowser.setText("Current working directory is " +
self.working_directory + "\n")
else:
self.working_directory = QFileDialog.getExistingDirectory(
self, "Select a working_directory", self.working_directory)
self.textBrowser.setText("Current working directory is " +
self.working_directory + "\n")
def initialize_configuration_fun(self):
"""Create file to save settings
This function will add the configuration_file to self
"""
if self.working_directory:
configuration_file_name, ok = QInputDialog.getText(
self, "Initialize configuration",
"Please name the configuration file:", QLineEdit.Normal,
"configuration_file.json")
if ok:
self.configuration_file = os.path.join(
self.working_directory, configuration_file_name)
with open(self.configuration_file, 'w') as configuration_file:
config = {
"data_loading": {},
"feature_engineering": {},
"machine_learning": {},
"model_evaluation": {},
}
config = json.dumps(config, indent=4)
configuration_file.write(config)
config_message = "Configuration file is " + self.configuration_file
self.textBrowser.setText(config_message)
else:
QMessageBox.warning(
self, 'Warning',
'Please choose a working directory first! (press button at the top left corner)'
)
def load_configuration_fun(self):
"""Load configuration
"""
if self.working_directory:
working_directory = self.working_directory
else:
working_directory = os.getcwd()
self.configuration_file, filetype = QFileDialog.getOpenFileName(
self, "Select configuration file", working_directory,
"Text Files (*.json);;All Files (*);;")
# Read configuration_file if already selected
if self.configuration_file != "":
with open(self.configuration_file, 'rb') as config:
# TODO:
"""<!-- The above is a description of an error in a Python program, formatted
for a Web browser because the 'cgitb' module was enabled. In case you
are not reading this in a Web browser, here is the original traceback:
Traceback (most recent call last):
File "D:\python\lib\site-packages\eslearn\GUI\easylearn_main_run.py", line 202, in load_configuration_fun
self.configuration = config.read()
UnicodeDecodeError: 'gbk' codec can't decode byte 0xad in position 177: illegal multibyte sequence
"""
self.configuration = config.read()
# Check the configuration is valid JSON, then transform the configuration to dict
# If the configuration is not valid JSON, then give configuration and configuration_file to ""
try:
self.configuration = json.loads(self.configuration)
self.textBrowser.setText("Configuration file is " +
self.configuration_file)
except json.decoder.JSONDecodeError:
QMessageBox.warning(
self, 'Warning',
f'{self.configuration_file} is not valid JSON')
self.configuration_file = ""
else:
QMessageBox.warning(self, 'Warning',
'Configuration file was not selected')
def data_loading_fun(self):
"""This function is called when data_loading button is clicked.
Then, this function will process the data loading.
"""
self.data_loading = EasylearnDataLoadingRun(self.working_directory)
self.data_loading.show()
def feature_engineering_fun(self):
"""This function is called when feature_engineering button is clicked.
Then, this function will process the feature_engineering.
"""
self.feature_engineering = EasylearnFeatureEngineeringRun(
self.working_directory)
self.feature_engineering.show()
def machine_learning_fun(self):
"""This function is called when machine_learning button is clicked.
Then, this function will process the data loading.
"""
self.machine_learning = EasylearnMachineLearningRun(
self.working_directory)
self.machine_learning.show()
def model_evaluation_fun(self):
"""This function is called when model_evaluation button is clicked.
Then, this function will process the model evaluation.
"""
self.model_evaluation = EasylearnModelEvaluationRun(
self.working_directory)
self.model_evaluation.show()
def run_fun(self):
"""This function is called when data_loading button is clicked.
Then, this function will process the data loading in another thread [class Run].
"""
which_ml_type_dict = {
"Classification": Classification,
"Regression": Regression,
}
baseml = BaseMachineLearning(self.configuration_file)
if self.configuration_file == "":
self.textBrowser.setText(
"You have to specify a configuration file by loading a configuration file at the upper left (Project Initialization)\n"
)
raise ValueError("You have to specify a configuration file\n")
else:
print('Running...\n')
self.textBrowser.setText(
f"configuration is in {self.configuration_file}\n")
baseml.get_all_inputs()
ml_type = baseml.machine_learning_type_[0]
out_dir = self.working_directory if self.working_directory else os.path.dirname(
self.configuration_file)
model = which_ml_type_dict[ml_type](
configuration_file=self.configuration_file, out_dir=out_dir)
model.main_run()
model.run_statistical_analysis()
# self.run = Run(which_ml_type_dict[ml_type], self.configuration_file, out_dir)
# self.run.start()
print(
"#" * 10 +
"Congratulations! Your machine learning task is finished successfully!"
+ "#" * 10 + "\n")
def closeEvent(self, event):
"""This function is called when exit icon of the window is clicked.
This function make sure the program quit safely.
"""
# Set qss to make sure the QMessageBox can be seen
reply = QMessageBox.question(self, 'Quit', "Are you sure to quit?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if reply == QMessageBox.Yes:
event.accept()
else:
event.ignore()
def closeEvent_button(self, event):
"""This function is called when quit button is clicked.
This function make sure the program quit safely.
"""
# Set qss to make sure the QMessageBox can be seen
reply = QMessageBox.question(self, 'Quit', "Are you sure to quit?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if reply == QMessageBox.Yes:
QCoreApplication.quit()