partimat-删除班级干线的积分

时间:2019-01-17 11:38:10

标签: r predict lda

我正在做线性判别分析。训练数据集有两个变量和〜100行,还有4个类。我想绘制预测边界(在原始变量的平面上)。我用klaR::partimat来做。

到目前为止,我有:

klaR::partimat(class ~  v1+ v2, data = dataset0, method = "lda", gs = NA), 

这给了我 Partiton Plot

我想从图中删除四个黑色圆圈(代表每个班级的主要部分)。我该怎么办?

或者,如何为每个点赋予不同的颜色?

1 个答案:

答案 0 :(得分:0)

参数col.mean完成任务。如果要从图中删除均值,则解决方案是设置col.mean = NA

partimat(Species ~ Sepal.Length + Petal.Length, data = iris, 
    method = "lda", gs = NA, col.mean = NA)

enter image description here

另一个选项是设置自定义颜色集。该颜色集的长度应等于类别的数量:

partimat(Species ~ Sepal.Length + Petal.Length, data = iris, 
    method = "lda", gs = NA, col.mean = c("darkblue", "forestgreen", "darkred"))

enter image description here