【SVM】支持向量机实例合集

基于Java的SVM（支持向量机）实例合集

以下是一个基于Java的SVM（支持向量机）实例合集，包含核心代码示例和应用场景说明。这些例子基于流行的机器学习库（如LIBSVM、Weka、JSAT）实现。

数据准备与加载

使用LIBSVM格式加载数据集：

// 加载LIBSVM格式数据
svm_problem prob = new svm_problem();
prob.l = dataSize; // 样本数量
prob.x = new svm_node[dataSize][]; // 特征向量
prob.y = new double[dataSize]; // 标签

// 示例数据填充（实际应从文件读取）
for(int i=0; i<dataSize; i++){
    prob.x[i] = new svm_node[featureSize];
    for(int j=0; j<featureSize; j++){
        prob.x[i][j] = new svm_node();
        prob.x[i][j].index = j+1;
        prob.x[i][j].value = Math.random(); // 示例随机值
    }
    prob.y[i] = i%2; // 二分类标签
}

模型训练

使用LIBSVM训练二分类模型：

svm_parameter param = new svm_parameter();
param.svm_type = svm_parameter.C_SVC; // C-SVM分类器
param.kernel_type = svm_parameter.RBF; // RBF核
param.gamma = 0.5; // 核函数参数
param.C = 1; // 惩罚系数
param.eps = 0.01; // 停止标准

svm_model model = svm.svm_train(prob, param);

交叉验证

执行k折交叉验证：

double[] target = new double[prob.l];
svm.svm_cross_validation(prob, param, 5, target); // 5折交叉验证

// 计算准确率
int correct = 0;
for(int i=0; i<prob.l; i++)
    if(target[i] == prob.y[i]) correct++;
System.out.println("Accuracy: "+(100.0*correct/prob.l)+"%");

文本分类

使用Weka实现文本分类：

// 加载ARFF格式文本数据
DataSource source = new DataSource("text_data.arff");
Instances data = source.getDataSet();
data.setClassIndex(data.numAttributes()-1);

// 配置SVM分类器
LibSVM svm = new LibSVM();
svm.setKernelType(new SelectedTag(LibSVM.KERNELTYPE_RBF, LibSVM.TAGS_KERNELTYPE));

// 训练与评估
Evaluation eval = new Evaluation(data);
eval.crossValidateModel(svm, data, 10, new Random(1));
System.out.println(eval.toSummaryString());

回归问题

使用SVR进行回归预测：