R(朴素贝叶斯分类器)中该语句的含义

时间:2017-05-03 20:55:24

标签: r naivebayes

我正在浏览我们教授为我们提供的代码,用于创建朴素贝叶斯分类器。请注意,我们没有使用一些内置包;而是为了学习目的自己写作。

教授使用的一个陈述让我感到困惑:

t = (Xtrain[,11] == c);

其中Xtrain是我们用来构造分类器的数据集。我想我明白了Xtrain[,11] == c的作用,但我没有得到的是t的作业。有人可以让我知道它的作用和原因吗?

编辑:

以下是他用来训练分类器的代码:

X = read.csv("naive_bayes_binary.csv");
tnum = nrow(X)/2;  
Xtrain = X[1:tnum,];  # the data we construct the classifier from
p = matrix(0,3,10);  #  p[c,j] = P(x_j = 1 | Y = c)
prior = rep(0,3);  # will be prior probs
n = rep(0,3);  # will be class counts
for (c in 1:3) {
    t = (Xtrain[,11] == c);    ### What is this?
    n[c] = sum(t);
    for (j in 1:10) {
        p[c,j] = sum(Xtrain[t,j] == 1)/n[c]  
    # empirical prob that jth feat = 1 for cth class
    }
}
prior = n/tnum;  # the prior probabilities of the classes

2 个答案:

答案 0 :(得分:1)

正如我在评论中提到的,t是逻辑的向量,表示c中的值等于X[,11]。如果您对向量t求和,则会得到出现次数(True1False0)。

这是一个小例子:

## 10 classes
n <- rep(0,10)

# class number of interest
c <- 7

# data vector (in OP's example a column)
X11 <- sample(1:10,100,replace = T)

X11
      [1]  2  7  5 10  4  5  1  7  4  4  1  8  1  5  7  1 10  2  6  9 10  4  3  2  2  8  7 10  3  2  5  3 10  4  8  2  2  8  6  2  5  2
     [43]  1  4  9  3  3  4  9  7  5 10 10  9  6 10  9  8  7  9  8  2  1  1  4  5  3 10  4  9 10  3 10  1  7 10  6  8  3  1  9  5  5  2
     [85]  9  9  1  9  3  3  3 10  5  3  3  2  7  4  3 10


# vector of logicals
t <- X11 == c

t
  [1] FALSE  TRUE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE
 [22] FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
 [43] FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE
 [64] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
 [85] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE

# assign number of occurrences
n[c] <- sum(t)

n的输出显示8次出现:

n
 [1] 0 0 0 0 0 0 8 0 0 0

答案 1 :(得分:0)

语句Xtrain[,11] == c返回一个TRUE或FALSE的布尔向量。如果Xtrain[,11]等于值c,则向量中的值将为TRUE,否则为0。然后将此向量分配给t

所以t可能如下所示:

FALSE
FALSE
TRUE
TRUE
FALSE
TRUE

或者其他什么。同样,当TRUE的相应行等于Xtrain[,11]时,会出现c值,该行在循环中设置为1,2或3。

此变量t显然在语句n[c] = sum(t);中稍后使用。当布尔值涉及算术时,它们被隐式转换为1(对于TRUE)和0(对于FALSE)。因此,基本n[c]将在输入Xtrain[,11]中包含1,2或3的计数。

这是一个完整的例子。假设Xtrain(Xtrain[,11])的第11列看起来像:

2
3
1
1
2
1

对于c = 1,t将如下所示:

FALSE
FALSE
TRUE
TRUE
FALSE
TRUE

所以n [c] = 3。

对于c = 2,t将如下所示:

TRUE
FALSE
FALSE
FALSE
TRUE
FALSE

所以n [2] = 2。

对于c = 3,t将如下所示:

FALSE
TRUE
FALSE
FALSE
FALSE
FALSE

所以n [3] = 1。