Source code for gui.core.SaveTrainedModel

import pickle

from PyQt5 import QtWidgets

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


# from gui.core.MainWindow import MainWindow



class SaveTrainedModel(Ui_Form, Modules):


    def setupUi(self, Form):
        super().setupUi(Form)
        Modules.setupUi(self, Form)




    def get_widget(self):
        return self.groupBox




    def on_savePushButton_clicked(self, lineEdit):
        self.filename, _filter = QtWidgets.QFileDialog.getSaveFileName(
            None,
            "Specify save file name",
            self.outpath,
            "(*.tram)"
        )
        lineEdit.setText(self.filename)
        if lineEdit.text() == "":
            lineEdit.setText("*.tram")




    def connectWidgets(self):
        self.setComboBox(self.choosemodel_comboBox, self.modelkeys)
        self.savePushButton.clicked.connect(
            lambda: self.on_savePushButton_clicked(self.saveLineEdit)
        )




    def run(self):
        modelkey = self.choosemodel_comboBox.currentText()
        filename = self.saveLineEdit.text()

        # infolist[0] holds the modelkey
        # infolist[1] holds the model
        # infolist[2] holds the xvar info
        # infolist[3] holds the yvar info

        infolist = []
        infolist.append(modelkey)
        infolist.append(self.models[modelkey])
        infolist.append(self.model_xvars[modelkey])
        infolist.append(self.model_yvars[modelkey])
        try:
            print(filename)
            with open(filename, 'wb') as fp:
                pickle.dump(infolist, fp)
        except Exception as e:
            print("Could not save your trained model: ", e)




if __name__ == "__main__":
    import sys

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