4

原来我想要的格式称为“SVM-Light”，并在此处描述http://svmlight.joachims.org/。

我有一个数据框，我想将其转换为格式如下的文本文件：

output featureIndex:featureValue ... featureIndex:featureValue

例如：

t = structure(list(feature1 = c(3.28, 6.88), feature2 = c(0.61, 1.83
), output = c("1", "-1")), .Names = c("feature1", "feature2", 
"output"), row.names = c(NA, -2L), class = "data.frame")

t
#   feature1 feature2 output
# 1     3.28     0.61      1
# 2     6.88     1.83     -1

会成为：

1 feature1:3.28 feature2:0.61
-1 feature1:6.88 feature2:1.83

到目前为止我的代码：

nvars = 2
l = array("row", nrow(t))
for(i in(1:nrow(t)))
{
    l = t$output[i]

    for(n in (1:nvars))
    {
        thisFeatureString = paste(names(t)[n], t[[names(t)[n]]][i], sep=":")
        l[i] = paste(l[i], thisFeatureString)
    }
}

但我不确定如何完成并将结果写入文本文件。此外，代码可能效率不高。

是否有执行此操作的库函数？例如，这种输出格式对于 Vowpal Wabbit 来说似乎很常见。

4

2 回答 2

2

我找不到现成的解决方案，尽管svm-light数据格式似乎被广泛使用。

这是一个可行的解决方案（至少在我的情况下）：

############### CONVERT DATA TO SVM-LIGHT FORMAT ##################################
# data_frame MUST have a column 'target'
# target values are assumed to be -1 or 1
# all other columns are treated as features
###################################################################################
ConvertDataFrameTo_SVM_LIGHT_Format <- function(data_frame)
{
    l = array("row", nrow(data_frame)) # l for "lines"
    for(i in(1:nrow(data_frame)))
    {
        # we start each line with the target value
        l[i] = data_frame$target[i]

        # then append to the line each feature index (which is n) and its 
        # feature value (data_frame[[names(data_frame)[n]]][i])
        for(n in (1:nvars))
        {
            thisFeatureString = paste(n, data_frame[[names(data_frame)[n]]][i], sep=":")
            l[i] = paste(l[i], thisFeatureString)
        }
    }

    return (l)
}
###################################################################################
于 2014-06-10T14:11:13.793 回答
1

如果您不介意在输出中没有列名，我认为您可以使用一个简单apply的方法来做到这一点：

apply(t, 1, function(x) paste(x, collapse=" "))
#[1] "3.28 0.61 1"  "6.88 1.83 -1"

并且要将输出中的出现顺序调整为函数的输出，您可以执行以下操作：

apply(t[c(3, 1, 2)], 1, function(x) paste(x, collapse=" "))
#[1] "1 3.28 0.61"  "-1 6.88 1.83"
于 2014-06-10T14:20:30.280 回答