在做多元逻辑回归进行分类问题时,经常需要将某一个分类转化成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