使用libsvm评估字符串数组

时间:2013-08-29 07:39:52

标签: java machine-learning svm libsvm

我使用java libsvm,我创建了我的模型,现在我想用字符串数组来评估它,我已经为双打数组做了

但我不知道如何评估字符串数组,我希望如果有人帮助我

提前感谢,

此代码使用双精度数组

private double evaluate(Neighbor features, svm_model model) {
    svm_node[] nodes = new svm_node[features.getDocument().getAttributes().size()];
    Iterator<String> iteraitor = features.getDocument().getAttributes().keySet().iterator();
    int i = 0;
    while (iteraitor.hasNext()) {
        svm_node node = new svm_node();
        node.index = i;
        String key = (String) iteraitor.next();
        node.value = (features.getDocument().getAttributes().get(key));
        nodes[i] = node;
        i++;
    }
int totalClasses = 2;
    int[] labels = new int[totalClasses];
    svm.svm_get_labels(model, labels);

    double[] prob_estimates = new double[totalClasses];
    double v = svm.svm_predict_probability(model, nodes, prob_estimates);

    for (int j = 0; j < totalClasses; j++) {
        System.out.print("(" + labels[j] + ":" + prob_estimates[j] + ")");
    }
    // System.out.println("(Actual:" + features[0] + " Prediction:" + v + ")");

    return v;
}

1 个答案:

答案 0 :(得分:2)

SVM是一种在R^n空间中工作的几何分类器,它接受仅数值值。要在字符串上使用SVM,您必须执行以下操作之一:

  • 将字符串/文档转换为某些特征表示(数字),如tfidf等。
  • 使用string-based kernel(如wordnet kernel等),这将自行介绍数字要素空间

libsvm允许使用预先计算的内核值,所以如果你实现了一些内核K(x_i,x_j),你只需要提供libsvm内核矩阵,其中K[i,j]=K(x_i,x_j)(其中{ {1}}是x_i培训示例。)