问题描述

在使用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 中即可

示例如下:
在这里插入图片描述

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐