Skip to content

liuslnlp/plume

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

plume

plume,一个轻量级机器学习库。包含常见机器学习算法的Python实现。

解决依赖

  • Python >= 3.6
  • NumPy
  • SciPy

快速入门

对于训练集X,标签集合y,算法clf,测试集合X_test,使用clf.fit(X, y)训练数据,使用clf.predcit(X_test)对测试集进行预测。

Input:

from plume.utils import plot_decision_boundary
from plume.svm import SVC

X, y = sklearn.datasets.make_moons(200, noise=0.20)
y = 2 * y - 1
clf = SVC(C=3, kernel='rbf')
clf.fit(X, y)
plot_decision_boundary(clf.predict, X, y, 'Support Vector Machine')

Output:

img