如何在 Python 中的图表上显示 R 平方值

Question

我是 Python 初学者，所以这可能比我想的更明显。我正在使用 Matplotlib 通过神经网络以图形方式呈现我的预测数据与实际数据。我能够计算 r 平方并绘制我的数据，但现在我想结合图表本身的值，该值随每个新的运行.

而变化

我的神经网络至少使用 4 个不同的输入，并给出一个输出。这是我的结束代码：

y_predicted = model.predict(X_test)

这是我计算 R2 的方式：

# Using sklearn
from sklearn.metrics import r2_score
print r2_score(y_test, y_predicted)

这是我的图表：

fig, ax = plt.subplots()
ax.scatter(y_test, y_predicted)
ax.plot([y.min(), y.max()], [y.min(), y.max()], 'k--', lw=4)
ax.set_xlabel('Actual')
ax.set_ylabel('Predicted')
#regression line
y_test, y_predicted = y_test.reshape(-1,1), y_predicted.reshape(-1,1)
ax.plot(y_test, LinearRegression().fit(y_test, y_predicted).predict(y_test))

plt.show()

它给出了类似于所附图表的内容，每次我更改时期、层数或数据类型等时，R2 都会变化。红色是我的回归线，稍后我将对其进行标记。由于 R2 是一个函数，我不能简单地使用图例或文本代码。

我也想显示MSE

谁能帮帮我？

Graph

Answer 1

如果我没理解错的话，你想在图中显示R2。您可以将其添加到图表标题中：

ax.set_title('R2: ' + str(r2_score(y_test, y_predicted)))

之前 plt.show()

Answer 2

如果您想将 r 平方值放在图形本身上，就像 Microsoft Excel 中的图形一样，您可以调用 annotate 方法：

ax.annotate("r-squared = {:.3f}".format(r2_score(y_test, y_predicted)), (0, 1))

或

plt.annotate("r-squared = {:.3f}".format(r2_score(y_test, y_predicted)), (0, 1))

第一个参数是您希望放置在图表上的文本，第二个参数是该文本左下角的位置。如果您想添加另一行，例如 MSE，您可以将“\n”和您希望的任何文本附加到第一个参数。请记住，(x, y) 元组是根据数据缩放的，因此如果您的数据范围不是 [0, 1]，并且您想将文本放在左上角，您可能需要更改来自 (0, 1).

的第二个参数

有关详细信息，请参阅 https://matplotlib.org/3.3.2/api/_as_gen/matplotlib.axes.Axes.annotate.html or https://matplotlib.org/3.3.2/api/_as_gen/matplotlib.pyplot.annotate.html

如何在 Python 中的图表上显示 R 平方值

How to display R-squared value on my graph in Python

python

matplotlib

legend

mse

neural-network