如何为Yellowbrick的PredictionError绘图设置X轴与Y轴标签？

方法1：初始化时直接指定标签（推荐） #

创建PredictionError可视化器时，通过x_label和y_label参数直接定义轴标签，这种方式不会被后续绘图流程覆盖：

from yellowbrick.regressor import PredictionError

# 初始化可视化器时传入标签参数
visualizer = PredictionError(LL_dt, x_label="Measured", y_label="Predicted")
visualizer.fit(X_train, y_train)
visualizer.score(X_test, y_test)
visualizer.poof()

方法2：在poof()后修改标签 #

如果需要在绘图完成后调整标签，可在调用poof()之后设置，此时默认标签已绘制完成，修改不会被覆盖：

from yellowbrick.regressor import PredictionError
import matplotlib.pyplot as plt

visualizer = PredictionError(LL_dt)
visualizer.fit(X_train, y_train)
visualizer.score(X_test, y_test)
visualizer.poof()

# 在poof之后设置自定义标签
visualizer.ax.set_xlabel("Measured")
visualizer.ax.set_ylabel("Predicted")
# 手动刷新画布确保修改生效
plt.show()

原方法无效的原因 #

你之前在score()之后、poof()之前设置标签，但PredictionError的poof()方法会自动生成默认轴标签（如"True Values"和"Predicted Values"），这会覆盖你提前设置的内容，导致修改无效。

内容的提问来源于stack exchange，提问作者Ningthoujam Jibanchand