工业4.0大数据竞赛 - 制造业质量控制

这是天池众智平台上的一道赛题：链接，本方案取得了第一名的成绩，解决方案和代码现整理开源。

解决方案

工业4.0大数据竞赛-技术方案总结

代码说明

data

存放原始的数据文件，包括：

• 训练数据，draft_data_train.csv, param_data_train.csv, timevarying_param_train.csv
• 旧测试数据，draft_data_test.csv, param_data_test.csv, timevarying_param_test.csv
• 新测试数据，draft_data_test_new.csv, param_data_test_new.csv, timevarying_param_test_new.csv

数据分析与特征提取

• plot_scatter.py，画散点图

• feature_extract.py,提取特征，在data目录下生成feature目录，存放特征文件

classification

• xgb_gt98.py，训练xgboost分类器，判断product_no的key_index是否大于0.98
• xgb_lt92.py，训练xgboost分类器，判断product_no的key_index是否小于0.92

regression

• old_testset目录

  • all目录
    • all_xgb.py，使用了所有特征训练的xgboost，做了特征选择
    • all_dart.py，使用了所有特征训练的dart，做了特征选择
  • half目录
    • xgb.py，使用了加工进度50%之前的特征训练的xgboost
  • draft目录
    • dart.py，使用draft_param特征训练的dart
  • gen_submission.py，生成提交文件

• new_testset目录

  • dp_xgb.py，使用draft_param和param两种特征训练的xgboost
  • dp_dart.py，使用draft_param和param两种特征训练的dart
  • dp_rf.py，使用draft_param和param两种特征训练的rf
  • d_xgb.py,使用draft_param特征训练的xgboost
  • d_rf.py,使用draft_param特征训练的rf
  • gen_submission.py，生成提交文件，融合了旧测试数据的结果。根据题目要求，分draft、half、all三种预测。

• post_process.py，使用分类模型的预测结果，对回归预测的结果进行后处理

recommend

• recommend.py,推荐三组工艺可调参数的预设值
• recommend_every_product.py,针对特定的工艺不可调整参数，对工艺可调参数进行推荐