132 lines
4.5 KiB
Python
132 lines
4.5 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
Step8 面板 - 机器学习预测
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from PyQt5.QtWidgets import (
|
|||
|
|
QWidget, QVBoxLayout, QGroupBox, QFormLayout,
|
|||
|
|
QPushButton, QCheckBox, QComboBox, QLineEdit, QMessageBox,
|
|||
|
|
QFileDialog,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
from src.gui.components.custom_widgets import FileSelectWidget
|
|||
|
|
from src.gui.styles import ModernStylesheet
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Step8Panel(QWidget):
|
|||
|
|
"""步骤8：机器学习预测"""
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self.init_ui()
|
|||
|
|
|
|||
|
|
def init_ui(self):
|
|||
|
|
layout = QVBoxLayout()
|
|||
|
|
|
|||
|
|
# 采样光谱CSV文件（用于独立运行）
|
|||
|
|
self.sampling_csv_file = FileSelectWidget(
|
|||
|
|
"采样光谱CSV:",
|
|||
|
|
"CSV Files (*.csv);;All Files (*.*)"
|
|||
|
|
)
|
|||
|
|
layout.addWidget(self.sampling_csv_file)
|
|||
|
|
|
|||
|
|
# 模型目录（用于独立运行）
|
|||
|
|
self.models_dir_file = FileSelectWidget(
|
|||
|
|
"模型目录:",
|
|||
|
|
"Directories;;All Files (*.*)"
|
|||
|
|
)
|
|||
|
|
self.models_dir_file.label.setText("模型目录:")
|
|||
|
|
self.models_dir_file.browse_btn.clicked.disconnect()
|
|||
|
|
self.models_dir_file.browse_btn.clicked.connect(self.browse_models_dir)
|
|||
|
|
layout.addWidget(self.models_dir_file)
|
|||
|
|
|
|||
|
|
# 参数设置
|
|||
|
|
params_group = QGroupBox("预测参数")
|
|||
|
|
params_layout = QFormLayout()
|
|||
|
|
|
|||
|
|
self.metric = QComboBox()
|
|||
|
|
self.metric.addItems(['test_r2', 'test_rmse', 'test_mae'])
|
|||
|
|
params_layout.addRow("模型选择指标:", self.metric)
|
|||
|
|
|
|||
|
|
self.prediction_column = QLineEdit()
|
|||
|
|
self.prediction_column.setText("prediction")
|
|||
|
|
params_layout.addRow("预测列名:", self.prediction_column)
|
|||
|
|
|
|||
|
|
params_group.setLayout(params_layout)
|
|||
|
|
layout.addWidget(params_group)
|
|||
|
|
|
|||
|
|
# 输出路径
|
|||
|
|
self.output_file = FileSelectWidget(
|
|||
|
|
"输出路径:",
|
|||
|
|
"CSV Files (*.csv);;All Files (*.*)"
|
|||
|
|
)
|
|||
|
|
layout.addWidget(self.output_file)
|
|||
|
|
|
|||
|
|
# 启用步骤
|
|||
|
|
self.enable_checkbox = QCheckBox("启用此步骤")
|
|||
|
|
self.enable_checkbox.setChecked(True)
|
|||
|
|
layout.addWidget(self.enable_checkbox)
|
|||
|
|
|
|||
|
|
# 独立运行按钮
|
|||
|
|
self.run_btn = QPushButton("独立运行此步骤")
|
|||
|
|
self.run_btn.setStyleSheet(ModernStylesheet.get_button_stylesheet('success'))
|
|||
|
|
self.run_btn.clicked.connect(self.run_step)
|
|||
|
|
layout.addWidget(self.run_btn)
|
|||
|
|
|
|||
|
|
layout.addStretch()
|
|||
|
|
self.setLayout(layout)
|
|||
|
|
|
|||
|
|
def browse_models_dir(self):
|
|||
|
|
"""浏览模型目录"""
|
|||
|
|
dir_path = QFileDialog.getExistingDirectory(self, "选择模型目录", "")
|
|||
|
|
if dir_path:
|
|||
|
|
self.models_dir_file.set_path(dir_path)
|
|||
|
|
|
|||
|
|
def get_config(self):
|
|||
|
|
"""获取配置"""
|
|||
|
|
config = {
|
|||
|
|
'metric': self.metric.currentText(),
|
|||
|
|
'prediction_column': self.prediction_column.text(),
|
|||
|
|
}
|
|||
|
|
sampling_csv_path = self.sampling_csv_file.get_path()
|
|||
|
|
if sampling_csv_path:
|
|||
|
|
config['sampling_csv_path'] = sampling_csv_path
|
|||
|
|
models_dir = self.models_dir_file.get_path()
|
|||
|
|
if models_dir:
|
|||
|
|
config['models_dir'] = models_dir
|
|||
|
|
output_path = self.output_file.get_path()
|
|||
|
|
if output_path:
|
|||
|
|
config['output_path'] = output_path
|
|||
|
|
return config
|
|||
|
|
|
|||
|
|
def set_config(self, config):
|
|||
|
|
"""设置配置"""
|
|||
|
|
if 'metric' in config:
|
|||
|
|
idx = self.metric.findText(config['metric'])
|
|||
|
|
if idx >= 0:
|
|||
|
|
self.metric.setCurrentIndex(idx)
|
|||
|
|
if 'prediction_column' in config:
|
|||
|
|
self.prediction_column.setText(config['prediction_column'])
|
|||
|
|
if 'sampling_csv_path' in config:
|
|||
|
|
self.sampling_csv_file.set_path(config['sampling_csv_path'])
|
|||
|
|
if 'models_dir' in config:
|
|||
|
|
self.models_dir_file.set_path(config['models_dir'])
|
|||
|
|
if 'output_path' in config:
|
|||
|
|
self.output_file.set_path(config['output_path'])
|
|||
|
|
|
|||
|
|
def run_step(self):
|
|||
|
|
"""独立运行步骤8"""
|
|||
|
|
sampling_csv_path = self.sampling_csv_file.get_path()
|
|||
|
|
models_dir = self.models_dir_file.get_path()
|
|||
|
|
if not sampling_csv_path:
|
|||
|
|
QMessageBox.warning(self, "输入错误", "请选择采样光谱CSV文件！")
|
|||
|
|
return
|
|||
|
|
if not models_dir:
|
|||
|
|
QMessageBox.warning(self, "输入错误", "请选择模型目录！")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
main_window = self.window()
|
|||
|
|
if hasattr(main_window, 'run_single_step'):
|
|||
|
|
config = {'step8': self.get_config()}
|
|||
|
|
main_window.run_single_step('step8', config)
|