最新消息:热烈庆祝IT小记上线!

Liblinear的使用

 Ubuntu下安装:http://pkgs.org/ubuntu-11.04/ubuntu-universe-i386/liblinear-tools_1.6+dfsg-1_i386.deb.html
然后软件中心查找liblinear的附加组件,添加Development libraries and header files for liblinear

Libsvm和Liblinear的使用经验谈 

http://blog.csdn.net/interxuxing/article/details/7607112

    LibsvmLiblinear都是国立台湾大学的Chih-Jen Lin博士开发的,Libsvm主要是用来进行非线性svm 分类器的生成,提出有一段时间了,而Liblinear则是去年才创建的,主要是应对large-scale的data classification,因为linear分类器的训练比非线性分类器的训练计算复杂度要低很多,时间也少很多,而且在large scale data上的性能和非线性的分类器性能相当,所以Liblinear是针对大数据而生的。


   两者都是一个跨平台的通用工具库,支持windows/linux/mac os,代码本身是c++写的,同时也有matlab,python,java,c/c++扩展接口,方便不同语言环境使用,可以说是科研和企业人员的首选!像我这样在学校的一般用matlab/c++,而我同学在百度则主要用的是python/c++,所以只是各自侧重不一样,但所使用的核心还是其svm库。


   以上Libsvm和Liblinear的主页上都有windows下的binary文件下载,zip,tar格式都有,解压后,找到matlab子文件目录,参看里面的readme文件,需要在matlab中进入此目录,运行make.m文件,matlab会根据本机默认的c/c++编译器将.c文件生成为.mexw32文件(由于我是32位操作系统,此处为mexw32,对于64位os,则对应为mexw64),提供matlab下能使用的接口。之后生成了这些.mexw32文件复制到你自己的matlab工程根目录中,就可以在matlab文件中调用libsvm/liblinear库中的函数了~


http://blog.sina.com.cn/s/blog_5bd2cb260100ev25.html 这位网友对libsvm在matlab中的使用说明的很详细,可以参考下。


有关Liblinear和Libsvm各自的优势可以归纳如下:

1.libsvm用来就解决通用典型的分类问题

2.liblinear主要为大规模数据的线性模型设计

  • it can be able to handle large-scaled dataset 可以用来处理大规模的数据
  • it runs really faster than libsvm because it doesn't have to compute thekernel for any two points 由于采用线性核,所以不需要计算kernel value,速度更快
  • trust region method for optimization looks new for machine learning people 

以下为一位网友采用liblinear进行数据分类的实验性能说明“

”今天试用了以下liblinear,速度很快(快到我没有想到), 
我的实验数据: 
训练集:21504 * 1500(1500是样本的数量,21504是维度) 
测试集:21504 * 2985 
速度用秒来衡量,20次实验总共不到2分钟。 
 
同样的问题我用了libsvm实验速度上相差太大,libsvm实验5次,每次将近10分钟,时间是其次,发现一个问题就是,libsvm比liblinear的结果相差1个百分点,没有读liblinear的文章,不知道问题出在那个地方,libsvm我直接用的默认参数,线性模型。这样必然引起一个问题,如果我想评价线性模型和非线性模型的性能,我不可能一个用liblinear一个用libsvm,如果两个都用libsvm,报告的性能肯定有一些问题。 
 
所以如果你的问题维度很大(线性模型就有非常好的性能),不妨考虑liblinear. “



大致看了一下libsvm和liblinear的说明文档,发现一个问题就是在线性问题上两者的目标函数就不一样,所以性能上的差异是正常的,应该说如果优化同一样的目标函数两者性能应该会差不多,但是速度很明显,liblinear快很多。 
 
对于什么时候用线性模型的问题,我想上面的我举的例子用linear classifier就比较好,非线性分类不一定比线性分类器好,尤其是在样本及其有限,同时特征维度很高的情况下,因为样本有限的情况下,kernel map通常不准确,很有可能错误地划分类别空间,可能造成比线性模型更差的结果。 

 

说到scale,我建议不要用libsvm里自带的scale,因为一旦使用这个工具,它就会把原来稀疏的数据,变成非稀疏的格式,这样不但会生成非常大的数据文件,而且liblinear对稀疏数据快速处理的优势就不能体现出来了。因此,要scale,就自己写一个,以保持原来稀疏的格式

liblinear的好处就是速度快,尤其是对稀疏的特征。缺点就是太吃内存了。10G的数据量需要接近50G的内存,数据量再大就没法做了 。


另外,还有一个经常提到的svm库SVM-per:http://www.cs.cornell.edu/people/tj/svm_light/svm_perf.html 是康奈尔大学的人设计的。好像对计算机硬件的性能要求比liblinear要低...有做图像处理的人使用这个svm-per代替liblinear。


另外,对于多分类问题以及核函数的选取,以下经验规则可以借鉴:

  • 如果如果特征数远远大于样本数的情况下,使用线性核就可以了.
  • 如果特征数和样本数都很大,例如文档分类,一般使用线性核, LIBLINEAR比LIBSVM速度要快很多.
  • 如果特征数远小于样本数,这种情况一般使用RBF.但是如果一定要用线性核,则选择LIBLINEAR较好,而且使用-s 2选项。

对于多分类问题:

对于15类场景来说,每类100幅训练图像,如果直接训练一个15类的multi-class classifier,则训练文件的Label值取1~15,wi标记不用指定(default 1)。如果对于每个类单独训练一个分类器,这样就把这个类的100幅图像作为正样本(假设Label=1),而其余所有的训练图像作为负样本(共1400幅,假设Label=-1),由此可以看出正负样本不平衡,因此应该制定wi选项,具体地可以指定-w1 14,-w-1 1( 1是缺省值),在cross validation的时候首先指定-wi,然后通过grid.py来确定(C,g)的最优值。实际做实验的时候,可以具体比较一下两种情况到底有多大差别

 

Libsvm与Liblinear源码剖析与封装(一)


简介

因为项目的需要,我们需要同时使用Libsvm与Liblinear,并将其封装起来做成统一调用形式,目前软件已经完成,名称为Tmsvm-基于SVM的文本挖掘系统
本文就来分析一下Libsvm与Liblinear的异同点以及封装这两个软件需要注意的事宜。

关于Libsvm的源码分析,上海交通大学模式识别实验室曾经有过这方面的工作,可以从网上下载。本文也结合他之前的工作,对libsvm的剖析做一些完善。目前关于Liblinear这方面的源码分析较少,因此我们就着手分析liblinear在具体实现上一些细节。

本文所分析所对应的libsvm的版本为3.0,liblinear的版本为1.8,开发语言为Java。对于Libsvm的Java版本,原作者用Java重新实现,而不是调用C++的动态链接库。而liblinear的Java版本因为原作者只提供了一个Jar包,并没有提供源代码,所以我们使用的是一个叫做Benedikt的人实现的,其源代码可在http://www.bwaldvogel.de/liblinear-java/ 进行下载。虽然是别人实现,但具体算法上和原作者基本相同。这也不影响本文对这两者进行一个综合的比较。

异同

关于Libsvm,台湾大学林智仁这样描述它:"LIBSVM is an integrated software for support vector classification, (C-SVC, nu-SVC), regression (epsilon-SVR, nu-SVR) and distribution estimation (one-class SVM). It supports multi-class classification."即Libsvm是一个整合了支持向量机(C-SVC, nu-SVC)、回归、分布估计(one-class SVM)的软件。并且支持多类别的分类。而对于LIblinear,官网上是这样介绍的:”LIBLINEAR is a linear classifier for data with millions of instances and features“,即主要专门为百万级别的数据和特征实现的线性分类器。

他们两个都是用来做分类的,相对来说Libsvm应用的范围较广, 而Liblinear主要用于处理大数据量的训练过程。在什么样的情况下,该选择Liblinear而不是Libsvm呢?作者给出几点建议:

  1.  当你面对海量的数据时,这里的海量通常是百万级别以上。海量数据分为两个层次:样本数量和特征的数量。
  2. 使用线性和非线性映射训练模型得到相近的效果。
  3. 对模型训练的时间效率要求较高。

在这类情况下,建议你使用Liblinear,而不是libsvm。文本分类是最典型的例子,文本分类的样本量非常多,而且特征的维度也是很高,从几千-几百万的数量级,因此在做文本方面的分类时最好选择liblinear。作者给出一个例子,对比liblinear与libsvm训练效果与时间效率。数据总共包含20,242样本,每个样本都包含47,236 个特征。

[plain] view plaincopy
  1. % time libsvm-2.85/svm-train -c 4 -t 0 -e 0.1 -m 800 -v 5 rcv1_train.binary  
  2. Cross Validation Accuracy = 96.8136%  
  3. 345.569s  
  4. % time liblinear-1.21/train -c 4 -e 0.1 -v 5 rcv1_train.binary  
  5. Cross Validation Accuracy = 97.0161%  
  6. 2.944s  

代码风格

因为不是同一个人缩写,所以在代码风格上可以看出,原作者充满了c的风味,因为连类的名字都是用 svm_model 这样的风格,liblinear的作者显然在这这些细节上上更像Java的风格。这里我们首先来剖析一下两者的代码风格:

下面要比较的两个类为特征向量中的节点类,Libsvm定义为svm_node,而Liblinear中定义为FeatureNode。即如果一个特征向量x={0.02,0.05,0,0.03},存储时就使用一个包含5 个svm_node 的数组来存储此4 维向量

1 2 3 4 -1
0.02 0.05 0 0.03

如果有一个节点的value为0,程序中是不会存储他的,上面的这个表只是为了能清晰的表示。

Libsvm

[java] view plaincopy
  1. public class svm_node implements java.io.Serializable  
  2. {  
  3.     public int index;  
  4.     public double value;  
  5. }  

Liblinear

[java] view plaincopy
  1. public class FeatureNode{  
  2.   
  3.     public final int index;  
  4.     public double    value;  
  5.   
  6.     public FeatureNode( final int index, final double value ) {  
  7.         if (index < 0throw new IllegalArgumentException("index must be >= 0");  
  8.         this.index = index;  
  9.         this.value = value;  
  10.     }  
  11.   
  12.     /** 
  13.      * @since 1.9 
  14.      */  
  15.     public int getIndex() {  
  16.         return index;  
  17.     }  
  18.   
  19.     /** 
  20.      * @since 1.9 
  21.      */  
  22.     public double getValue() {  
  23.         return value;  
  24.     }  
  25.   
  26.     /** 
  27.      * @since 1.9 
  28.      */  
  29.     public void setValue(double value) {  
  30.         this.value = value;  
  31.     }  
  32.   
  33.     @Override  
  34.     public int hashCode() {  
  35.         final int prime = 31;  
  36.         int result = 1;  
  37.         result = prime * result + index;  
  38.         long temp;  
  39.         temp = Double.doubleToLongBits(value);  
  40.         result = prime * result + (int)(temp ^ (temp >>> 32));  
  41.         return result;  
  42.     }  
  43.   
  44.     @Override  
  45.     public boolean equals(Object obj) {  
  46.         if (this == obj) return true;  
  47.         if (obj == nullreturn false;  
  48.         if (getClass() != obj.getClass()) return false;  
  49.         FeatureNode other = (FeatureNode)obj;  
  50.         if (index != other.index) return false;  
  51.         if (Double.doubleToLongBits(value) != Double.doubleToLongBits(other.value)) return false;  
  52.         return true;  
  53.     }  
  54.   
  55.     @Override  
  56.     public String toString() {  
  57.         return "FeatureNode(idx=" + index + ", value=" + value + ")";  
  58.     }  
  59. }  

从上述代码,我们理解可以看出若干端倪:

  1. 在类命名上,Libsvm的作者沿袭了C的风格,开头小写,两个单词之间用”_“连接。而Liblinear在继承了Java命名的一贯风格,开头大写,第二个单词也大写。
  2. 在类的定义上,Libsvm更像是C中的Struct,没有构造函数,只给出了成员变量,而且都是public型,也没有相应的Getter和Setter。而liblinear则遵循了Java类封装的原则。
  3. Liblinear除了定义构造函数、Getter、Setter外,还定义了hashCode()和equals()。如果将FeatureNode作为HashMap的Key,定义这两个函数至关重要。

好了,上面这些只是开胃小菜,接下来我们将会深入的剖析Libsvm与Liblinear的异同,以及封装他们的细节。


Libsvm与Liblinear源码浅析与封装(二)

封装Libsvm与Liblinear


开篇我们基于以下场景:

       假设我们已经用libsvm或者是Liblinear训练了一个模型,现在需要读入该模型并基于该模型对一个样本进行预测,返回预测的类标签以及相应的隶属度分数。

 从封装的角度我们需要考虑一下几个问题:

  1.  如何规范输入,使得读入的数据在Libsvm与Liblinear上都可以直接运行
  2. 系统如何识别读入的模型的类别,并根据模型类别,选择Libsvm或者是Liblinear中合适的函数进行预测
  3. 如何预测该样本对应的标签以及得分

系统输入

因为我们对单个样本进行预测,所以这里直接定义单个节点的格式即可。对于多个样本,则可以循环调用。
在上一篇文章中,我们看到虽然Libsvm与Liblinear在向量节点的定义上有所差别,但是其中的成员变量还是一致的,因此为了统一,我们只需要重新定义一个新的类。
[java] view plaincopy
  1. public class SvmNode {  
  2.     /** 封装的Libsvm与Liblinear的节点格式**/  
  3.     private final int index;  
  4.     private double value;  
  5.   
  6.     public SvmNode( final int index, final double value ) {  
  7.         if (index < 0throw new IllegalArgumentException("index must be >= 0");  
  8.         this.index = index;  
  9.         this.value = value;  
  10.     }  
  11.       
  12.     /**以下Getter和Setter、hashCode、equals函数省略**/  
  13.           
  14. }  

模型读入

我们首先来剖析一下Libsvm与Liblinear生成的模型的文件格式


从上述文件中,我们看到两者的模型文件存在很大的差异,因此我们只需要写一个解析函数直接读取模型的格式来判断到底属于那种模型。

最简单的形式就是直接通过第一行来判断,因为其模型的文件都是固定的,所以这种最简单的方式也是很有效的。

[java] view plaincopy
  1. /** 
  2.  * 读入模型文件。通过文件中的格式来判断到底属于Libsvm还是liblinear 
  3.  * @param model_filename 模型文件保存的路径及名称 
  4.  */  
  5. public void load_model(String model_filename){  
  6.     BufferedReader fp;  
  7.     try {  
  8.         InputStream in = new FileInputStream(model_filename);                 
  9.         fp = new BufferedReader(new InputStreamReader(in,"UTF-8"));  
  10.   
  11.         String line = null;  
  12.         try {  
  13.             while((line = fp.readLine())!=null){              
  14.                 String[] arg = whitespace.split(line);  
  15.                 //通过判断文件中每行第一个字符是否包含"svm_type"或者是"solver_type"  
  16.                 //来判断是Libsvm还是Liblinear  
  17.                 if(arg[0].equals("svm_type")){  
  18.                     svm_type = "libsvm";  
  19.                     svmModel = svm.svm_load_model(model_filename);  
  20.                     nr_class = svmModel.nr_class;  
  21.                     linearModel = null;  
  22.                     break;  
  23.                 }  
  24.                 if(arg[0].equals("solver_type")){  
  25.                     svm_type="liblinear";  
  26.                     svmModel = null;  
  27.                     linearModel = Linear.loadModel(new File(model_filename));  
  28.                     nr_class = linearModel.getNrClass();  
  29.                     break;  
  30.                 }  
  31.             }  
  32.         } catch (IOException e) {  
  33.             // TODO Auto-generated catch block  
  34.             e.printStackTrace();  
  35.         }  
  36.     } catch (UnsupportedEncodingException e1) {  
  37.         // TODO Auto-generated catch block  
  38.         e1.printStackTrace();  
  39.     } catch (FileNotFoundException e) {  
  40.         // TODO Auto-generated catch block  
  41.         e.printStackTrace();  
  42.     }  
  43. }  

多分类策略

这里我们研究一下Libsvm与Liblinear的多分类策略,虽然为多分类,但其实也适用于二分类。
对于Libsvm来说,采用为One-against-one的策略,而Liblinear则采用为One-against-rest的策略,关于这两类分类策略的解释,可以看下图:


我们接下来通过源代码具体分析一下:

首先来看一下Libsvm的模型训练部分:

[java] view plaincopy
  1. /** 
  2.  * Libsvm的模型训练函数 
  3.  * @param prob 输入的问题 
  4.  * @param param 输入的参数 
  5.  * @return 训练好的模型 
  6.  */  
  7. public static svm_model svm_train(svm_problem prob, svm_parameter param)  
  8. {  
  9.     /**为了代码的简洁性,此处删除了部分**/  
  10.       
  11.     /** 
  12.      * 此处省略了部分代码:这里的代码主要做: 
  13.      * 1、统计类别总数,同时记录类别的标号,统计每个类的样本数目 
  14.      * 2、计算权重C 
  15.      * 3、初始化nozero数组,便于统计SV 
  16.      * 4、初始化概率数组 
  17.     */  
  18.     svm_model model = new svm_model();  
  19.     model.param = param;  
  20.   
  21.     // 将属于同一个类别的数据组织起来  
  22.     svm_group_classes(prob,tmp_nr_class,tmp_label,tmp_start,tmp_count,perm);  
  23.       
  24.     //采用one-against-one进行分类  
  25.     int p = 0;  
  26.     //定义决策函数数组,将每一个而分类存储起来  
  27.     decision_function[] f = new decision_function[nr_class*(nr_class-1)/2];  
  28.     for(i=0;i<nr_class;i++)  
  29.         for(int j=i+1;j<nr_class;j++)  
  30.             svm_problem sub_prob = new svm_problem();  
  31.       
  32.     //计算第i个类与第j个类的起始位置与数目。  
  33.             int si = start[i], sj = start[j];  
  34.             int ci = count[i], cj = count[j];  
  35.             sub_prob.l = ci+cj;  
  36.     //将第i个类的标签定义为+1,第j个类的标签定义为-1  
  37.             for(k=0;k<ci;k++)  
  38.             {  
  39.                 sub_prob.x[k] = x[si+k];  
  40.                 sub_prob.y[k] = +1;  
  41.             }  
  42.             for(k=0;k<cj;k++)  
  43.             {  
  44.                 sub_prob.x[ci+k] = x[sj+k];  
  45.                 sub_prob.y[ci+k] = -1;  
  46.             }  
  47.     //对第i个类与第j个类采用二分类策略,训练模型  
  48.             f[p] = svm_train_one(sub_prob,param,weighted_C[i],weighted_C[j]);  
  49.             ++p;  
  50.         }  
  51.     /** 
  52.      * 此处省略了部分代码:这里的代码主要做: 
  53.      * 1、统计一下nozero,如果nozero已经是真,就不变,如果为假,则改为真 
  54.      * 2、输出模型,主要是填充svm_model 
  55.     */  
  56.     return model;  
  57. }  

然后再来看一下其模型预测部分:

[java] view plaincopy
  1. /** 
  2.  * Libsvm模型预测部分,预测该样本的类标签以及属于该类的分值。 
  3.  * 这里采用的方法为投票策略 
  4.  * @param model 已训练好的Libsvm模型 
  5.  * @param x 一个待预测的样本向量 
  6.  * @param dec_values 保存预测结果的向量,其维度为k*(k-1)/2 
  7.  * @return 
  8.  */  
  9. public static double svm_predict_values(svm_model model, svm_node[] x, double[] dec_values)  
  10. {  
  11.       
  12.     //对一些变量进行初始化  
  13.     int i;  
  14.     int nr_class = model.nr_class;  
  15.     int l = model.l;  
  16.   
  17.     double[] kvalue = new double[l];  
  18.     //根据模型中的支持向量以及内核函数计算数值  
  19.     for(i=0;i<l;i++)  
  20.         kvalue[i] = Kernel.k_function(x,model.SV[i],model.param);  
  21.   
  22.     int[] start = new int[nr_class];  
  23.     start[0] = 0;  
  24.     for(i=1;i<nr_class;i++)  
  25.         start[i] = start[i-1]+model.nSV[i-1];  
  26.       
  27.     //初始化vote向量  
  28.     int[] vote = new int[nr_class];  
  29.     for(i=0;i<nr_class;i++)  
  30.         vote[i] = 0;  
  31.   
  32.     //依次计算k*(k-1)/2个二分类模型的预测值  
  33.     int p=0;  
  34.     for(i=0;i<nr_class;i++)  
  35.         for(int j=i+1;j<nr_class;j++)  
  36.         {  
  37.             double sum = 0;  
  38.             int si = start[i];  
  39.             int sj = start[j];  
  40.             int ci = model.nSV[i];  
  41.             int cj = model.nSV[j];  
  42.           
  43.         //根据决策函数计算每一个二分类对样本预测的分值  
  44.             int k;  
  45.             double[] coef1 = model.sv_coef[j-1];  
  46.             double[] coef2 = model.sv_coef[i];  
  47.             for(k=0;k<ci;k++)  
  48.                 sum += coef1[si+k] * kvalue[si+k];  
  49.             for(k=0;k<cj;k++)  
  50.                 sum += coef2[sj+k] * kvalue[sj+k];  
  51.             sum -= model.rho[p];  
  52.             //dec_values存储的就是第p个二分类模型对样本的预测值  
  53.             dec_values[p] = sum;                      
  54.   
  55.         //根据dec_values  值的正负决定给那个类别投票  
  56.             if(dec_values[p] > 0)  
  57.                 ++vote[i];  
  58.             else  
  59.                 ++vote[j];  
  60.             p++;  
  61.         }  
  62.       
  63.     //遍历vote向量,找出具有最大投票数的所对应类别即为所预测的类标签。  
  64.     //而dec_values则存储了k*(k-1)/2个二分类模型的预测值  
  65.     int vote_max_idx = 0;  
  66.     for(i=1;i<nr_class;i++)  
  67.         if(vote[i] > vote[vote_max_idx])  
  68.             vote_max_idx = i;  
  69.   
  70.     return model.label[vote_max_idx];  
  71.   
  72. }  

接下来我们分析一下Liblinear的模型训练与模型预测部分

模型训练部分:
[java] view plaincopy
  1. /** 
  2.  * Liblinear模型训练部分 
  3.  * @param prob 输入的问题 
  4.  * @param param 输入的参数 
  5.  * @return 返回训练好的模型 
  6.  */public static Model train(Problem prob, Parameter param) {  
  7.   
  8.     /** 
  9.      * 此处省略了部分代码:这里的代码主要做: 
  10.      * 1、统计类别总数,同时记录类别的标号,统计每个类的样本数目 
  11.      * 2、将属于相同类的样本分组,连续存放 
  12.      * 3、计算权重C 
  13.      * 4、初始化nozero数组,便于统计SV 
  14.      * 5、初始化概率数组 
  15.     */  
  16.       
  17.     //此处可以看出需要寻k个模型  
  18.     for (int i = 0; i < nr_class; i++) {  
  19.         int si = start[i];  
  20.         int ei = si + count[i];  
  21.           
  22.         //重建子数据集,样本的特征不变,但样本的类别要改为+1/-1  
  23.         int k = 0;  
  24.         for (; k < si; k++)  
  25.             sub_prob.y[k] = -1;  
  26.         for (; k < ei; k++)  
  27.             sub_prob.y[k] = +1;  
  28.         for (; k < sub_prob.l; k++)  
  29.             sub_prob.y[k] = -1;  
  30.         //训练子数据集svm_train_one  
  31.         train_one(sub_prob, param, w, weighted_C[i], param.C);  
  32.   
  33.         for (int j = 0; j < n; j++)  
  34.             model.w[j * nr_class + i] = w[j];  
  35.     }  
  36.   
  37.   
  38.     /** 
  39.      * 此处省略了部分代码:这里的代码主要做: 
  40.      * 1、统计一下nozero,如果nozero已经是真,就不变,如果为假,则改为真 
  41.      * 2、输出模型,主要是填充svm_model 
  42.     */  
  43.       
  44.     return model;  
  45. }  

模型预测部分:

[java] view plaincopy
  1. /** 
  2.  * 模型预测部分,根据训练好的模型对样本进行预测。 
  3.  * @param model 已训练好的模型 
  4.  * @param x 样本的向量 
  5.  * @param dec_values 每个模型预测分值 
  6.  * @return 预测的类标签 
  7.  */  
  8. public static int predictValues(Model model, Feature[] x, double[] dec_values) {  
  9.       
  10.     //变量的初始化  
  11.     int n;  
  12.     if (model.bias >= 0)  
  13.         n = model.nr_feature + 1;  
  14.     else  
  15.         n = model.nr_feature;  
  16.   
  17.     double[] w = model.w;  
  18.   
  19.     int nr_w;  
  20.     if (model.nr_class == 2 && model.solverType != SolverType.MCSVM_CS)  
  21.         nr_w = 1;  
  22.     else  
  23.         nr_w = model.nr_class;  
  24.       
  25.     for (int i = 0; i < nr_w; i++)  
  26.         dec_values[i] = 0;  
  27.       
  28.     //计算每个模型预测分值  
  29.     for (Feature lx : x) {  
  30.         int idx = lx.getIndex();  
  31.         // the dimension of testing data may exceed that of training  
  32.         if (idx <= n) {  
  33.             for (int i = 0; i < nr_w; i++) {  
  34.                 dec_values[i] += w[(idx - 1) * nr_w + i] * lx.getValue();  
  35.             }  
  36.         }  
  37.     }  
  38.     //根据预测的分值来计算所属的标签,和Libsvm所不同的是:  
  39.     //这里采用的方式是比较那个模型预测分值大。  
  40.     if (model.nr_class == 2)  
  41.         return (dec_values[0] > 0) ? model.label[0] : model.label[1];  
  42.     else {  
  43.         int dec_max_idx = 0;  
  44.         for (int i = 1; i < model.nr_class; i++) {  
  45.             if (dec_values[i] > dec_values[dec_max_idx]) dec_max_idx = i;  
  46.         }  
  47.         return model.label[dec_max_idx];  
  48.     }  
  49. }  

综上所述,Libsvm采用的为One-against-one策略,Liblinear采用的One-against-rest策略。最后都能返回预测的类标签。但是对于由于他们采用的策略不同,导致其在预测的分值上不同。所以接下来我们将会对预测的分值进行归一化处理。

模型结果

模型会返回两个结果:label和score,其中label即其预测的标签。而score是该样本属于该类的隶属度,分值越大,代表属于该类的置信度越大。具体的计算方式则是根据:

其中k为所有支持判别类得个数,n为所有类别个数,si 为所有支持判别类的分数。这样就相当于把两个软件计算出来的分数进行了统一,而且经过公式的计算,会将分数的映射到[0,1]区间内,这样方便进行阈值控制。

代码为 

[java] view plaincopy
  1. /** 
  2.  * 返回具有最大投票数的标签所获得分数的总和 
  3.  * @param des_values 
  4.  */  
  5. public double sumPreValue(double[] des_values){  
  6.     int size=1;  
  7.     double init_score=0.0;  
  8.     int k=1;  
  9.     //对于Libsvm的类型,对其k*(k-1)/2个值进行计算  
  10.     if (this.svm_type.equals("libsvm")){  
  11.         int n = 1+(int)Math.sqrt(2.0*des_values.length+1);  
  12.         size = n-1;  
  13.         int [] vote = new int[n];  
  14.         double[] score = new double[n];  
  15.         int p =0;  
  16.         for(int i =0;i<n-1;i++){  
  17.             for(int j =i+1;j<n;j++){  
  18.                 if(des_values[p]>0){  
  19.                     vote[i]+=1;  
  20.                     score[i]+=Math.abs(des_values[p]);  
  21.                 }else{  
  22.                     vote[j]+=1;  
  23.                     score[j]+=Math.abs(des_values[p]);  
  24.                 }  
  25.                 p+=1;  
  26.             }  
  27.         }  
  28.         int max = 0;  
  29.         for(int i =1;i<n;i++)  
  30.             if(vote[i]>vote[max])  
  31.                 max = i;  
  32.         k = vote[max];  
  33.         init_score = score[max];  
  34.     }  
  35.     //对于Liblinear的类型,对其k个值进行计算  
  36.     if (this.svm_type.equals("liblinear")){  
  37.         int n =des_values.length;  
  38.         int max = 0;  
  39.         for(int i =1;i<n-1;i++){  
  40.             if(des_values[i]>des_values[max]){  
  41.                 max = i;  
  42.             }  
  43.         }  
  44.         size =1;  
  45.         k=1;  
  46.         init_score = des_values[max];  
  47.            
  48.     }  
  49.     return init_score/(2.0*k)+k/(2.0*size) ;  
  50.   
  51. }  


以Libsvm为例,说明一下分数的计算


总结

这样我们通过输入一个样本,根据读入的模型,就可以真正的对样本进行预测了。而且通过归一化两者的分值,可以很好的对结果进行阈值控制。更多详细的代码请见 https://code.google.com/p/tmsvm/


猜您喜欢

备案号:苏ICP备12016861号-4