多元逻辑回归 - 小技巧(一)

在做多元逻辑回归进行分类问题时，经常需要将某一个分类转化成vector，或者反过来的操作。比如一共有四种分类，那么第二种分类的vector就是：[0;1;0;0], 除了第二个元素为1外，其它的都为0.

所以问题为：给定一个vector，里边的值代表了是第几重分类：

比如y = [1;2;4]; 三个样本，第一个样本为分类1，第二个样本为分类2，第三个样本为分类4. 构造一个矩阵，样本分类的相应位为1，其它位为0. 变成：

[ 1 0 0 0;

  0 1 0 0;

  0 0 0 1;]

三个样本的四分类矩阵，怎么做呢？

octave/matlab里有一个概念，叫做逻辑数组，logical array，可以简单的写成y==i, i是样本分类的数值。

for i = 1:m,

  val(i,:) = (y == i);

 

那么反过来怎么做呢？比如我们predict的时候，算出来了分类矩阵为：

[ 0 1 0 0;

  0 1 0 0;

  0 0 0 1;]

我们想把它变成[2;2;4], 怎么做呢?可以用max函数，max不但能求最大值，还能求最大值的索引下标。

[v,p] = max(results, [], 2);
p就是我们想要的。不懂max用法可以help max;

原文:http://blog.csdn.net/hongchangfirst/article/details/81068661

作者:hongchangfirst

hongchangfirst的主页:http://blog.csdn.net/hongchangfirst