数学之路(3)-机器学习(3)-机器学习算法-余弦相似度(4)

本博客所有内容是原创，如果转载请注明来源

http://blog.youkuaiyun.com/myhaspl/

我们用下面三个待分类图片进行测试

ptest3.jpg

ptest1.jpg

ptest2.jpg

测试分类

http://blog.youkuaiyun.com/myhaspl
myhaspl@qq.com


正在处理中

ptest3.jpg属于第3类

ptest1.jpg属于第1类
ptest2.jpg属于第2类

>>>

分类效果不错,测试分类的python代码如下：

fn='ptest3.jpg'
testtz=np.array(readpic(fn))
simtz=testtz[0].tolist()+testtz[1].tolist()+testtz[2].tolist()
maxtz=0
nowi=0
for i in xrange(0,picflag):
    nowsim=get_cossimi(train_x[i],simtz)
    if nowsim>maxtz:
        maxtz=nowsim
        nowi=i        
print u'%s属于第%d类'%(fn,nowi+1)

fn='ptest1.jpg'
testtz=np.array(readpic(fn))
simtz=testtz[0].tolist()+testtz[1].tolist()+testtz[2].tolist()
maxtz=0
nowi=0
for i in xrange(0,picflag):
    nowsim=get_cossimi(train_x[i],simtz)
    if nowsim>maxtz:
        maxtz=nowsim
        nowi=i        
print u'%s属于第%d类'%(fn,nowi+1)
        
fn='ptest2.jpg'
testtz=np.array(readpic(fn))
simtz=testtz[0].tolist()+testtz[1].tolist()+testtz[2].tolist()
maxtz=0
nowi=0
for i in xrange(0,picflag):
    nowsim=get_cossimi(train_x[i],simtz)
    if nowsim>maxtz:
        maxtz=nowsim
        nowi=i        
print u'%s属于第%d类'%(fn,nowi+1)