一种基于不平衡数据集的PTM-WKNN分类方法和装置

摘要

本发明公开了一种基于不平衡数据集的PTM‑WKNN分类方法和装置。其中所述方法包括：获取测试集样本的多个最近邻的训练集样本，分别计算训练集样本最佳局部k值并选择最大的所述训练集样本最佳局部k值作为测试集样本最佳局部k值；计算所述测试集样本与所述测试集样本最佳局部k值个最近邻所述训练集样本之间的欧式距离，根据所述欧式距离计算所述训练集样本所占的权重，根据所述权重计算正类邻居权重之和以及负类邻居权重之和；根据所述正类邻居权重之和以及所述负类邻居权重之和生成所述测试集样本的分类结果。该方法和装置考虑了待分类样本的局部特性，又考虑到了每个近邻样本与待测样本的距离对样本分类的影响，具有更好的技术效果。