项目作者： WiseDoge

项目描述：
常见机器学习算法的Python实现

高级语言： Python

项目主页：

项目地址: git://github.com/WiseDoge/plume.git

创建时间： 2017-02-23T02:55:15Z
项目社区：https://github.com/WiseDoge/plume
开源协议：MIT License
关键词：
algorithm data-science datamining machine-learning-algorithms machinelearning python
下载

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:


