plt.scatter(test_target_tensor.numpy(), final_output.numpy())
时间: 2024-02-23 10:58:28 浏览: 61
python matplotlib:plt.scatter() 大小和颜色参数详解
5星 · 资源好评率100%
这行代码使用了Matplotlib库中的scatter函数,用于画散点图。其中,test_target_tensor和final_output都是Numpy数组,分别表示测试集的目标值和模型预测值。通过将这两个数组作为函数的输入,可以画出目标值和预测值之间的散点图。
```python
import matplotlib.pyplot as plt
# 画散点图
plt.scatter(test_target_tensor.numpy(), final_output.numpy())
# 设置图表标题和轴标签
plt.title('目标值 vs 预测值')
plt.xlabel('目标值')
plt.ylabel('预测值')
# 显示图表
plt.show()
```
在这个例子中,我们首先使用scatter函数画出散点图,然后设置了图表标题和轴标签,并用show函数显示图表。
阅读全文