正确预测的 R 可视化

Question

我已经训练了基于识别数字 0-9 的概率预测的 SVM 分类模型。 我有每个模型的概率可视化，数字 0 看起来像这样 - 概率数据在变量 prediction0

然后我训练了最终分类器并且我有 1423 个正确的观察结果（从 1499 开始）- 我有向量 c= 包含正确预测的数字

我需要做的是，什么时候在向量 c 中正确预测了 0，在这个 graf 上用红色标记那个点。如果它有帮助，我有“ck”包含每个测试样本的所有数字预测的概率，其中我得到最大概率，这是我的最终预测。

Answer 1

您可以使用 col 参数来完成此操作。我将使用 mtcars 数据集作为示例

plot(
  mpg~disp,
  data=mtcars,
  col=ifelse(mtcars$am==0,"red","blue")
)