为R中的dataframe列中的每个组创建模型

时间:2018-11-23 06:36:01

标签: r dataframe grouping naivebayes party

我想为列TN的每组值创建一个naiveBayes模型。 这是代码:

library(party)
airq <- subset(airquality, !is.na(Ozone))
ct <- ctree(Ozone ~ ., data = airq,  controls = ctree_control(maxsurrogate = 3))
airq$TN<-factor (ct@where)

这里是一个例子:

> tail (airq)
    Ozone Solar.R Wind Temp Month Day TN
1     7      49 10.3   69     9  24   2
2    14      20 16.6   63     9  25   5
3    30     193  6.9   70     9  26   3
4    14     191 14.3   75     9  28   2
5    18     131  8.0   76     9  29   5
6    20     223 11.5   68     9  30   3

我想为每个值建立一个NaiveBayes模型:

model[i] <- naiveBayes(Ozone ~ ., data = dat[i])

dat[1]将包含用于创建第一个模型的第1行和第4行(TN=2),dat[2]包含第二个模型的{TN=5)的第2行和第5行,{ {1}}包含第3行,并包含第三个模型(dat[3])。 所以最后,我将可以总结一下:对于TN中的每个唯一组值,我都会有其模型。 像这样:

TN=3

0 个答案:

没有答案