ML之LoR:基于LoR算法实现对非线性数据集点进行绘制决策边界

ML之LoR:基于LoR算法实现对非线性数据集点进行绘制决策边界


基于LoR算法实现对非线性数据集点进行绘制决策边界

1、查看数据集

import numpy as np
from sklearn.datasets import make_moons  #make_moons数据集可以生成一些非线性数据点
import matplotlib.pyplot as plt

# 手动生成一个随机的平面点分布,并画出来
np.random.seed(0)
X, y = make_moons(200, noise=0.20)
plt.scatter(X[:,0], X[:,1], s=40, c=y, cmap=plt.cm.Spectral)
plt.title('make_moons generating nonlinear data points')
plt.show()

2、设计代码

from sklearn.linear_model import LogisticRegressionCV

#查看LoR算法的分类效果
#因为线性切分,决策边界是直线,这条直线只能尽量让某一类点落在某一侧,但是永远做不到真的完全切分
clf = LogisticRegressionCV()
clf.fit(X, y)

#绘制一下决策边界
plot_decision_boundary(lambda x: clf.predict(x))
plt.title("Logistic Regression")
plt.show()

3、输出结果

(0)

相关推荐