decision_tree = tree.DecisionTreeClassifier(max_depth=6)decision_tree.fit(X_train, y_train)如何用散点图可视化数据划分的结果
时间: 2024-05-05 09:21:40 浏览: 105
要将数据划分的结果可视化为散点图,您需要执行以下步骤:
1. 预测训练集中的类别。
```python
y_pred = decision_tree.predict(X_train)
```
2. 将 X_train 和 y_pred 转换为 Pandas DataFrame。
```python
import pandas as pd
df = pd.DataFrame({'x': X_train[:, 0], 'y': X_train[:, 1], 'class': y_pred})
```
3. 使用 Seaborn 绘制散点图,其中颜色表示类别。
```python
import seaborn as sns
sns.scatterplot(x='x', y='y', hue='class', data=df)
```
完整的代码如下所示:
```python
import seaborn as sns
import pandas as pd
from sklearn import tree
# 创建决策树分类器
decision_tree = tree.DecisionTreeClassifier(max_depth=6)
# 拟合模型
decision_tree.fit(X_train, y_train)
# 预测训练集中的类别
y_pred = decision_tree.predict(X_train)
# 将 X_train 和 y_pred 转换为 Pandas DataFrame
df = pd.DataFrame({'x': X_train[:, 0], 'y': X_train[:, 1], 'class': y_pred})
# 使用 Seaborn 绘制散点图,其中颜色表示类别
sns.scatterplot(x='x', y='y', hue='class', data=df)
```
这样,您就可以将数据划分的结果可视化为散点图了。
阅读全文