BNlearn R错误“变量Variable1必须至少具有两个级别。”

时间:2019-02-28 09:40:34

标签: r bnlearn

尝试使用BNlearn创建BN,但我一直遇到错误;

Error in check.data(data, allowed.types = discrete.data.types) : variable Variable1 must have at least two levels.

这给我每个变量带来的错误,即使它们都是因素,并且具有多个级别,如您所见-在这种情况下,我的变量“模型”具有4个级别

由于我无法共享变量和数据集,因此我创建了一个小集合并为该数据集添加了相关代码。我遇到同样的问题。我知道我只共享2个变量,但是所有变量都得到相同的错误。

library(tidyverse)
library (bnlearn)
library(openxlsx)

DataFull <- read.xlsx("(.....)/test.xlsx", sheet = 1, startRow = 1, colNames = TRUE)
set.seed(600)
DataFull <- as_tibble(DataFull)

DataFull$Variable1 <- as.factor(DataFull$Variable1)
DataFull$TargetVar <- as.factor(DataFull$TargetVar)

DataFull <- na.omit(DataFull)
DataFull <- droplevels(DataFull)

DataFull <- DataFull[sample(nrow(DataFull)),]
Data <- DataFull[1:as.integer(nrow(DataFull)*0.70)-1,]
Datatest <- DataFull[as.integer(nrow(DataFull)*0.70):nrow(DataFull),]
nrow(Data)+nrow(Datatest)==nrow(DataFull)

FocusVar <- as.character("TargetVar")
BN.naive <- naive.bayes(Data, FocusVar) 

使用str(data),我可以看到该变量已经具有2个或更多级别:

  

str(数据)

Classes ‘tbl_df’, ‘tbl’ and 'data.frame':   27586 obs. of  2 variables:
 $ Variable1: Factor w/ 3 levels "Small","Medium",..: 2 2 3 3 3 3 3 3 3 3 ...
 $ TargetVar: Factor w/ 2 levels "Yes","No": 1 1 1 1 1 1 2 1 1 1 ...

链接到数据集:https://drive.google.com/open?id=1VX2xkPdeHKdyYqEsD0FSm1BLu1UCtOj9eVIVfA_KJ3g

1 个答案:

答案 0 :(得分:1)

bnlearn期望data.frame:不能与tibbles一起使用,因此通过省略行data.frame DataFull <- as_tibble(DataFull)。 >

示例

library(tibble)
library (bnlearn)

d <- as_tibble(learning.test)
hc(d)
  

check.data(x)中的错误:变量A必须至少具有两个级别。

尤其是bnlearn:::check.data

中的那一行
if (nlevels(x[, col]) < 2) 
      stop("variable ", col, " must have at least two levels.")

在标准data.frame中,learning.test[,"A"]返回一个向量,因此nlevels(learning.test[,"A"])可以按预期工作,但是,根据设计,您不能从tibbles提取这样的向量:{ {1}}仍然是d[,"A"])而不是向量,因此tbl_df无法正常工作,并返回零。