/ML_for_PID

Primary LanguageJupyter Notebook

ML_for_PID

主文件是pid.ipynb, Pi.dat和K.dat,Proton.dat是数据文件,分别记录了Pi,K,proton的飞行参数,每一行对应一个事例,每个事例11维数据。3种类型粒子每种1万共3万训练数据,以及每种粒子3000的测试数据。经过数据清洗以及预处理后,建立XGB,LGB,RF,SVC4种模型,通过调整参数得到最优分辨率后,利用Stacking进行融合,得出总体准确率约为97%,也利用ROC曲线以及精确率、召回率对不同模型的分类结果进行比较。准确率比传统方法高出约4%。

详情可以查看pid.ipynb文件