基于ml-knn的多标签分类

首先要区别多分类 和多标签的区别。

多分类问题:一个样本的类别是多个,比如判别一个未知水果,它可能是苹果,也可能是香蕉,也可能是西瓜。水果的种类的是多样的,但是一个样本只可能是其中一类。
多标签问题:一个样本包含多个类别。例如评价某个人和善,聪明,懒惰,即这个样本包含多个类别标签,既可以是A也可以是B,类别之间不互斥。

算法原理

改进版knn :ml-knn (multi-label knn)

  • 1.通过knn 算法寻找和样本最近的K个样本
  • 2.统计k个样本中每个类别的个数
  • 3.根据第二步的统计,采用 native bayes算法计算每个标签的概率
  • 4.输出类别概率

算法伪代码:

code

(1)-(13)步骤是native bayes 模型训练
步骤一:由(1)-(2) 统计类别L在样本中的概率。
s :参数,用于数据平滑,
m:是标准样本个数。
p(H^l_1):表示标签L的概率
\sum_{i=0}^my_x{_i}(l): 表示样本中标签L的个数。

步骤二:(3)-(13)计算每个样本的K个最近邻中,在样本标签为L的条件下,E^l_j(K个样本中有j个L标签的样本数量)的概率。以及在样本不是标签L的条件下,E^l_j的概率

步骤三:新样本估计
y_t = argmax_{b\{0,1\}} P(H^l_b)P(E^l_j|H^l_b)  
(这里native bayes中的分母省略计算,比较分子大小就够了)
其中:y_t指的是样本类别向量
伪代码中r_t就是样本每个类的概率。

算法简单实践:

MLkNN 来自于scikit-multilearn库
库安装:

pip install scikit-multilearn

数据读取

import scipy
import pandas as pd
from scipy.io import arff
data, meta = scipy.io.arff.loadarff('/app/jupyter_dir/data/yeast-train.arff')
df = pd.DataFrame(data)
df.head()
    Att1    Att2    Att3    Att4    Att5    Att6    Att7    Att8    Att9    Att10   ... Class5  Class6  Class7  Class8  Class9  Class10 Class11 Class12 Class13 Class14
0   0.093700    0.139771    0.062774    0.007698    0.083873    -0.119156   0.073305    0.005510    0.027523    0.043477    ... b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'
1   -0.022711   -0.050504   -0.035691   -0.065434   -0.084316   -0.378560   0.038212    0.085770    0.182613    -0.055544   ... b'0'    b'0'    b'1'    b'1'    b'0'    b'0'    b'0'    b'1'    b'1'    b'0'
2   -0.090407   0.021198    0.208712    0.102752    0.119315    0.041729    -0.021728   0.019603    -0.063853   -0.053756   ... b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'1'    b'1'    b'0'
3   -0.085235   0.009540    -0.013228   0.094063    -0.013592   -0.030719   -0.116062   -0.131674   -0.165448   -0.123053   ... b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'1'    b'1'    b'1'
4   -0.088765   -0.026743   0.002075    -0.043819   -0.005465   0.004306    -0.055865   -0.071484   -0.159025   -0.111348   ... b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'    b'0'

这里用yeast的数据做测试,可以看出样本数据中有14个类别。

数据切分

from sklearn.model_selection import train_test_split

X = df.iloc[:,0:103].values
y = df.iloc[:,103:117].values

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)

数据类型转换

import numpy as np
y_train = y_train.astype(np.float64)
y_test = y_test.astype(np.float64)

模型训练及预测

from skmultilearn.adapt import MLkNN
from sklearn.metrics import accuracy_score 
classifier = MLkNN(k=20)
# train
classifier.fit(X_train, y_train)
# predict
predictions = classifier.predict(X_test)

准确率统计

z = predictions.toarray()
[rows, cols] = y_test.shape
zzz = 0
for i in range(rows - 1):
    for j in range(cols - 1):
        if y_test[i, j] == z[i, j]:
            zzz += 1
zzz / (rows*cols) 

实验基于jupyter,下载:
链接:链接:https://pan.baidu.com/s/1SLo5nZjvrVrJwqgoEvOpoQ 密码:qge8

点个赞呗~老哥们。

参考文献

论文地址:http://cse.seu.edu.cn/PersonalPage/zhangml/files/ML-kNN%20a%20lazy%20learning%20approach%20to%20multi-label%20learning.pdf

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,723评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,003评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,512评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,825评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,874评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,841评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,812评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,582评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,033评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,309评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,450评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,158评论 5 341
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,789评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,409评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,609评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,440评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,357评论 2 352

推荐阅读更多精彩内容