如何创建模型以对特定组中的数据进行分类

时间:2018-02-20 06:18:59

标签: r machine-learning regression data-modeling

以下是我的数据集的样子:

A1  B1  C1  D1  E1  F1
A   1   10  100 2   0.5
A   2   20  120 4   0.7
A   3   30  300 5   0.8
A   4   2   200 1   0.9
B   5   5   115 1   1
B   6   100 100 1   0

我不熟悉建模。我必须选择一个模型,以便为每个A1选择最小值为F1的B1(例如:A)。任何人都可以建议我应该使用什么样的模型来实现同样的目标。

1 个答案:

答案 0 :(得分:1)

可以这样做

data <- data.frame(
     A1 = c("A", "A", "A", "A", "B", "B"),
     B1 = c(1, 2, 3, 4, 5, 6),
     C1 = c(10, 20, 30, 2, 5, 100),
     D1 = c(100, 120, 300, 200, 115, 100),
     E1 = c(2, 4, 5, 1, 1, 1),
     F1 = c(0.5, 0.7, 0.8, 0.9, 1, 0)
)

res <- c(by(data = data, INDICES = data$A1, FUN = function(x) x$B1[which.min(x$F1)], simplify = TRUE))
# A B 
# 1 6