当前位置:   article > 正文

支持向量机代码实现_.coef[0]

.coef[0]
  1. print(__doc__)
  2. # 导入相关的包
  3. import numpy as np
  4. import pylab as pl # 绘图功能
  5. from sklearn import svm
  6. # 创建 40 个点
  7. np.random.seed(0) # 让每次运行程序生成的随机样本点不变
  8. # 生成训练实例并保证是线性可分的
  9. # np._r表示将矩阵在行方向上进行相连
  10. # random.randn(a,b)表示生成 a 行 b 列的矩阵,且随机数服从标准正态分布
  11. # array(20,2) - [2,2] 相当于给每一行的两个数都减去 2
  12. X = np.r_[np.random.randn(20, 2) - [2, 2], np.random.randn(20, 2) + [2, 2]]
  13. # 两个类别 每类有 20 个点,Y 为 40 行 1 列的列向量
  14. Y = [0] * 20 + [1] * 20
  15. # 建立 svm 模型
  16. clf = svm.SVC(kernel='linear')
  17. clf.fit(X, Y)
  18. # 获得划分超平面
  19. # 划分超平面原方程:w0x0 + w1x1 + b = 0
  20. # 将其转化为点斜式方程,并把 x0 看作 x,x1 看作 y,b 看作 w2
  21. # 点斜式:y = -(w0/w1)x - (w2/w1)
  22. w = clf.coef_[0] # w 是一个二维数据,coef 就是 w = [w0,w1]
  23. a = -w[0] / w[1] # 斜率
  24. xx = np.linspace(-5, 5) # 从 -5 到 5 产生一些连续的值(随机的)
  25. # .intercept[0] 获得 bias,即 b 的值,b / w[1] 是截距
  26. yy = a * xx - (clf.intercept_[0]) / w[1] # 带入 x 的值,获得直线方程
  27. # 画出和划分超平面平行且经过支持向量的两条线(斜率相同,截距不同)
  28. b = clf.support_vectors_[0] # 取出第一个支持向量点
  29. yy_down = a * xx + (b[1] - a * b[0])
  30. b = clf.support_vectors_[-1] # 取出最后一个支持向量点
  31. yy_up = a * xx + (b[1] - a * b[0])
  32. # 查看相关的参数值
  33. print("w: ", w)
  34. print("a: ", a)
  35. print("support_vectors_: ", clf.support_vectors_)
  36. print("clf.coef_: ", clf.coef_)
  37. # 在 scikit-learin 中,coef_ 保存了线性模型中划分超平面的参数向量。形式为(n_classes, n_features)。若 n_classes > 1,则为多分类问题,(1,n_features) 为二分类问题。
  38. # 绘制划分超平面,边际平面和样本点
  39. pl.plot(xx, yy, 'k-')
  40. pl.plot(xx, yy_down, 'k--')
  41. pl.plot(xx, yy_up, 'k--')
  42. # 圈出支持向量
  43. pl.scatter(clf.support_vectors_[:, 0], clf.support_vectors_[:, 1],
  44. s=80, facecolors='none')
  45. pl.scatter(X[:, 0], X[:, 1], c=Y, cmap=pl.cm.Paired)
  46. pl.axis('tight')
  47. pl.show()

随机创建40个点,同时生成相应的矩阵且服从正态分布,将这些数据分成两部分。然后建立svm模型,通过计算获得斜点式方程的斜率和截距,带入x值获得直线方程,取最初和最后一个支持向量点,绘制出划分超平面边际平面和样本点,圈出支持向量。

输出结果:

  1. w: [0.90230696 0.64821811]
  2. a: -1.391980476255765
  3. support_vectors_: [[-1.02126202 0.2408932 ]
  4. [-0.46722079 -0.53064123]
  5. [ 0.95144703 0.57998206]]
  6. clf.coef_: [[0.90230696 0.64821811]]

w=clf.coef,a为求直线二维数据和斜率,support_vectors_为支持向量点

输出图表: 

通过svm预测划分类型

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/563656
推荐阅读
相关标签
  

闽ICP备14008679号