f6660518b7
重构主要管道代码，实现更完善的架构设计： - 添加统一配置管理系统，支持数据、模型、训练和可视化参数的集中管理 - 实现完整的日志记录系统，支持文件和控制台输出，提供结构化的运行日志 - 集成性能监控工具，包括执行时间、内存使用情况的实时跟踪和分析 - 新增模型检查点功能，支持模型和评估结果的自动保存与加载 - 添加数据验证器，确保数据完整性和预测结果的正确性 - 实现模型工厂模式，便于扩展和管理不同类型的机器学习模型 - 优化异常处理机制，提供专门的异常类型分类 - 更新依赖项，添加psutil库用于系统资源监控 输出文件管理改进： - 将评估结果和训练后的模型保存到outputs目录 - 添加日志文件到.gitignore，确保版本控制清洁 - 提供完整的模型评估指标对比分析
27 lines
638 B
JSON
27 lines
638 B
JSON
{
|
|
"Stacking Model": {
|
|
"rmse": 64.64642644074704,
|
|
"mae": 47.963500739758054,
|
|
"r2": 0.8930611738230659
|
|
},
|
|
"Ridge": {
|
|
"rmse": 20.258368087201287,
|
|
"mae": 16.172561165969487,
|
|
"r2": 0.9894984045767364
|
|
},
|
|
"XGBoost": {
|
|
"rmse": 77.03564948928194,
|
|
"mae": 58.84620600162227,
|
|
"r2": 0.848144731434714
|
|
},
|
|
"LightGBM": {
|
|
"rmse": 72.76199680324312,
|
|
"mae": 57.38448079041826,
|
|
"r2": 0.8645261150841663
|
|
},
|
|
"MLP": {
|
|
"rmse": 23.392914827573286,
|
|
"mae": 18.809170451107256,
|
|
"r2": 0.9859971948234603
|
|
}
|
|
} |