背景介绍

  • svm是一种有监督式机器学习,即需要事先特征及对应打标label录入系统,机器学习其规律。
  • 输入与输出有些像LR,只是算法原理不同导致输出结果不一样。

代码

  1. import numpy as np
  2. from sklearn.svm import SVC
  3. #训练集
  4. X = np.array([[-1, -1], [-2, -1], [1, 1], [2, 1]])
  5. #标签
  6. y = np.array([1, 1, 2, 2])
  7. clf = SVC(gamma='auto')
  8. clf.fit(X, y)
  9. #预测
  10. print(clf.predict([[-0.8, -1]]))

参考资料