将模型的二分类结果进行可视化导出。代码
时间: 2024-03-24 08:41:58 浏览: 50
二分类的数据集
以下是一个简单的示例代码,用于可视化二分类模型的结果:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import make_classification
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
# 生成数据集
X, y = make_classification(n_samples=1000, n_features=2, n_informative=2, n_redundant=0, n_clusters_per_class=1, random_state=42)
# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 训练模型
clf = LogisticRegression()
clf.fit(X_train, y_train)
# 预测测试集
y_pred = clf.predict(X_test)
# 可视化
plt.figure(figsize=(10, 8))
plt.scatter(X_test[:, 0], X_test[:, 1], c=y_pred)
plt.title("Binary Classification Results")
plt.xlabel("Feature 1")
plt.ylabel("Feature 2")
plt.show()
```
在上面的示例代码中,我们使用`make_classification`函数生成了一个包含1000个样本和2个特征的数据集,并使用Logistic回归模型进行二分类训练。最后,我们使用散点图可视化了测试集的结果,其中不同的颜色表示不同的类别。这个示例可以帮助你理解如何可视化二分类模型的结果。
阅读全文