决策树预测值真实值散点图代码
时间: 2023-07-09 13:19:58 浏览: 161
以下是一个示例代码,它使用 `scatter` 函数绘制决策树预测值和真实值的散点图:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
import matplotlib.pyplot as plt
# 加载数据集
iris = load_iris()
X = iris.data[:, 2:] # 取花瓣长度和宽度两个特征
y = iris.target
# 训练决策树模型
tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)
# 预测样本标签
y_pred = tree_clf.predict(X)
# 绘制散点图
plt.scatter(y, y_pred)
plt.xlabel('True Label')
plt.ylabel('Predicted Label')
plt.show()
```
这段代码首先训练了一个决策树模型,并使用训练数据集 `X` 对其进行预测,得到预测结果 `y_pred`。然后使用 `scatter` 函数绘制散点图,其中 `x` 参数表示真实值,`y` 参数表示预测值。这样可以通过散点图来观察预测值和真实值之间的关系,以评估决策树模型的性能。
相关问题
决策树对预测值和真实值做散点图代码
以下是使用Python代码绘制决策树预测值和真实值的散点图的示例:
```python
import matplotlib.pyplot as plt
import numpy as np
from sklearn.tree import DecisionTreeRegressor
# 训练数据集
X_train = np.array([[1], [2], [3], [4], [5], [6], [7], [8], [9], [10]])
y_train = np.array([2, 4, 6, 8, 10, 12, 14, 16, 18, 20])
# 测试数据集
X_test = np.array([[0.5], [1.5], [2.5], [3.5], [4.5], [5.5], [6.5], [7.5], [8.5], [9.5]])
# 创建决策树模型
tree = DecisionTreeRegressor(max_depth=2)
tree.fit(X_train, y_train)
# 对测试数据进行预测
y_pred = tree.predict(X_test)
# 绘制散点图
plt.scatter(X_test, y_pred, color='red', label='Predicted Values')
plt.scatter(X_test, tree.predict(X_test), color='blue', label='True Values')
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Decision Tree Regression')
plt.legend()
plt.show()
```
在这个示例中,我们使用了一个简单的训练数据集和测试数据集来创建和测试决策树回归模型。我们使用matplotlib库中的scatter函数来绘制散点图,其中红色点代表模型预测的值,蓝色点代表真实的值。通过比较这两组点的位置和分布,我们可以直观地看出模型的预测效果和准确性。
决策树模型预测出来的真实值和预测值用散点图画出来代码
可以使用 `scatter` 函数绘制散点图,其中真实值和预测值用不同的颜色来表示。以下是一个示例代码:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
import matplotlib.pyplot as plt
# 加载数据集
iris = load_iris()
X = iris.data[:, 2:] # 取花瓣长度和宽度两个特征
y = iris.target
# 训练决策树模型
tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)
# 预测样本标签
y_pred = tree_clf.predict(X)
# 绘制散点图
plt.scatter(X[:, 0], X[:, 1], c=y, cmap=plt.cm.get_cmap('viridis', 3), label='True Label')
plt.scatter(X[:, 0], X[:, 1], c=y_pred, cmap=plt.cm.get_cmap('viridis', 3), marker='x', s=50, label='Predicted Label')
plt.xlabel('Petal Length')
plt.ylabel('Petal Width')
plt.legend()
plt.show()
```
这段代码首先训练了一个决策树模型,并使用训练数据集 `X` 对其进行预测,得到预测结果 `y_pred`。然后使用 `scatter` 函数绘制散点图,其中 `c` 参数表示颜色映射,`cmap` 参数指定使用的颜色映射类型,`marker` 参数指定预测值使用的标记类型,`s` 参数指定标记的大小。利用 `label` 参数分别标注真实值和预测值的类别。
阅读全文