sklearn逻辑回归怎么可视化
时间: 2023-08-27 16:59:53 浏览: 369
要可视化Scikit-learn中的逻辑回归模型,您可以绘制决策边界和数据点的散点图。这可以帮助您直观地理解模型的分类能力。
下面是一个使用Scikit-learn和Matplotlib库可视化逻辑回归模型的示例:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
# 加载数据集
iris = load_iris()
X = iris.data[:, :2] # 只选择前两个特征以便绘图
y = iris.target
# 创建逻辑回归模型并拟合数据
clf = LogisticRegression(random_state=0)
clf.fit(X, y)
# 绘制散点图
plt.scatter(X[:, 0], X[:, 1], c=y, cmap=plt.cm.Set1, edgecolor='k')
plt.xlabel('Sepal length')
plt.ylabel('Sepal width')
# 绘制决策边界
x_min, x_max = X[:, 0].min() - .5, X[:, 0].max() + .5
y_min, y_max = X[:, 1].min() - .5, X[:, 1].max() + .5
xx, yy = np.meshgrid(np.linspace(x_min, x_max, 200), np.linspace(y_min, y_max, 200))
Z = clf.predict(np.c_[xx.ravel(), yy.ravel()])
Z = Z.reshape(xx.shape)
plt.contour(xx, yy, Z, levels=[0], linewidths=2)
plt.show()
```
上述代码将绘制一个散点图,其中数据点的颜色表示其所属的类别,同时还绘制了逻辑回归模型的决策边界。您可以通过调整特征选择、模型参数等来观察决策边界的变化。请注意,这只是一个简单的示例,具体的绘图方法可能因数据集和需求而有所不同。
阅读全文