R语言LDA模型报错Error in partimat.default(x, grouping, ...) : at least two classes required的原因及解决方法

问题描述

在使用LDA/QDA模型做预测后，使用partimat 函数做LDA的决策边界可视化，结果报错如下

Error in partimat.default(x, grouping, …) :
at least two classes required

原因

在调用命令 partimat( formula(lda.fit), data = tmp, method=“lda”, plot.matrix =TRUE) 时，data 这个argument 需要我们输入 factor 类型的class/group/Y，比如我们想用LDA模型判断一个公司会不会破产时，class/group/Y 就是“yes/no” 或者0/1，我们需要把这个字段从 numeric type 或者 character type 转变成 factor 类型

解决方法

把进行分类预测的class/group （比如0/1或者default/not）那一列字段用 as.factor 转成factor 格式，然后再输入到 partimat 函数的argument 中即可

示例如下：