SVD/SVD++实现推荐算法

奇异值分解(Singular Value Decomposition,以下简称SVD)是在机器学习领域广泛应用的算法,它不仅可以用于降维算法中的特征分解,还可以用于推荐系统,以及自然语言处理等领域。

我们首先回顾下特征值和特征向量的定义如下:

                                                                    Ax=λx

其中A是一个n×n的矩阵,x是一个n维向量,则我们说λ是矩阵A的一个特征值,而x是矩阵A的特征值λ所对应的特征向量。

  求出特征值和特征向量有什么好处呢? 我们可以将矩阵A特征分解。如果我们求出了矩阵A的n个特征值λ1λ2...λn,以及这n个特征值所对应的特征向量{w1,w2,...wn},那么矩阵A就可以用下式的特征分解表示:

A=WΣW1
其中W是这n个特征向量所张成的n×n维矩阵,而Σ为这n个特征值为主对角线的n×n维矩阵。

  一般我们会把W的这n个特征向量标准化,即满足||wi||2=1, 或者说wTiwi=1,此时W的n个特征向量为标准正交基,满足WTW=I,即WT=W1, 也就是说W为酉矩阵。

    这样我们的特征分解表达式可以写成A=WΣWT

    注意到要进行特征分解,矩阵A必须为方阵。那么如果A不是方阵,即行和列不相同时,我们还可以对矩阵进行分解吗?答案是可以,此时我们的SVD登场了。

  SVD的定义

    SVD也是对矩阵进行分解,但是和特征分解不同,SVD并不要求要分解的矩阵为方阵。假设我们的矩阵A是一个m×nm×n的矩阵,那么我们定义矩阵A的SVD为:

                                                       A=UΣV  

    其中U是一个m×m的矩阵,Σ是一个m×n的矩阵,除了主对角线上的元素以外全为0,主对角线上的每个元素都称为奇异值,V是一个n×n的矩阵。U和V都是酉矩阵,即满足UTU=I,VTV=I。下图可以很形象的看出上面SVD的定义:

    那么我们如何求出SVD分解后的U,Σ,V这三个矩阵呢?

    如果我们将A的转置和A做矩阵乘法,那么会得到n×nn×n的一个方阵ATA。既然ATA是方阵,那么我们就可以进行特征分解,得到的特征值和特征向量满足下式:

                                                                    (ATA)vi=λivi

    这样我们就可以得到矩阵ATA的n个特征值和对应的n个特征向量v了。将ATA的所有特征向量张成一个n×n的矩阵V,就是我们SVD公式里面的V矩阵了。一般我们将V中的每个特征向量叫做A的右奇异向量。

    如果我们将A和A的转置做矩阵乘法,那么会得到m×m的一个方阵AAT。既然AAT是方阵,那么我们就可以进行特征分解,得到的特征值和特征向量满足下式:

                                                               (AAT)ui=λiui

    这样我们就可以得到矩阵AAT的m个特征值和对应的m个特征向量u了。将AAT的所有特征向量张成一个m×m的矩阵U,就是我们SVD公式里面的U矩阵了。一般我们将U中的每个特征向量叫做A的左奇异向量。

    U和V我们都求出来了,现在就剩下奇异值矩阵Σ没有求出了。由于Σ除了对角线上是奇异值其他位置都是0,那我们只需要求出每个奇异值σ就可以了。

    我们注意到:

                                                              A=UΣVTAV=UΣVTVAV=UΣAvi=σiuiσi=Avi/ui

     这样我们可以求出我们的每个奇异值,进而求出奇异值矩阵Σ

             SVD的一些性质 

    上面几节我们对SVD的定义和计算做了详细的描述,似乎看不出我们费这么大的力气做SVD有什么好处。那么SVD有什么重要的性质值得我们注意呢?

    对于奇异值,它跟我们特征分解中的特征值类似,在奇异值矩阵中也是按照从大到小排列,而且奇异值的减少特别的快,在很多情况下,前10%甚至1%的奇异值的和就占了全部的奇异值之和的99%以上的比例。也就是说,我们也可以用最大的k个的奇异值和对应的左右奇异向量来近似描述矩阵。也就是说:

                                                   Am×n=Um×mΣm×nVTn×nUm×kΣk×kVTk×n

    其中k要比n小很多,也就是一个大的矩阵A可以用三个小的矩阵Um×k,Σk×k,VTk×n来表示。如下图所示,现在我们的矩阵A只需要灰色的部分的三个小矩阵就可以近似描述了。

    由于这个重要的性质,SVD可以用于PCA降维,来做数据压缩和去噪。也可以用于推荐算法,将用户和喜好对应的矩阵做特征分解,进而得到隐含的用户需求来做推荐。同时也可以用于NLP中的算法,比如潜在语义索引(LSI)。

以上转自:http://www.cnblogs.com/pinard/p/6251584.html

SVD协同过滤:

假设存在以下user和item的数据矩阵:

这是一个极其稀疏的矩阵,这里把这个评分矩阵记为R,其中的元素表示user对item的打分,“?”表示未知的,也就是要你去预测的,现在问题来了:如何去预测未知的评分值呢?从上面的SVD的性质:  Am×n=Um×mΣm×nVTn×nUm×kΣk×kVTk×n,可以得到:

一个m*n的打分矩阵R可以由分解的两个小矩阵U(m*k)V(k*n)的乘积来近似,即 R=UVT,k<=m,n 

将这种分解方式体现协同过滤中,即有:

          技术分享 (matrix factorization model,MF模型 )

 在这样的分解模型中,Pu代表用户隐因子矩阵(表示用户u对因子k的喜好程度),Qi表示电影隐因子矩阵(表示电影i在因子k上的程度)。

SVD推荐算法公式如下:

    技术分享

这里需要解释一下各个参数的含义:

对于电影评分实例,首先得到训练数据集 user_id,movie_id和rating,u表示打分矩阵中所有评分值的平均值,bi在这个公式中应该是一个参数值,而不是向量,可以这样理解,首先初始化一个代表item的向量bi,向量维度是item的个数,公式中的bi是指bi[movie_id],同理,bu代表bu[user_id],rui_hat 表示预测的评分值.

加入防止过拟合的 λ 参数,可以得到下面的优化函数:

     技术分享

利用随机梯度下降算法更新参数:

 代码体现:

# -*- coding: utf-8 -*-
"""
Created on Thu Apr 19 16:28:10 2018

@author: 
"""
import numpy as np  
import random  
import os 
  
class SVD:  
    def __init__(self,mat,K=20):  
        self.mat=np.array(mat)  
        self.K=K  
        self.bi={}  
        self.bu={}  
        self.qi={}  
        self.pu={}  
        self.avg=np.mean(self.mat[:,2])  
        for i in range(self.mat.shape[0]):  
            uid=self.mat[i,0]  
            iid=self.mat[i,1]  
            self.bi.setdefault(iid,0)  
            self.bu.setdefault(uid,0)  
            self.qi.setdefault(iid,np.random.random((self.K,1))/10*np.sqrt(self.K))  
            self.pu.setdefault(uid,np.random.random((self.K,1))/10*np.sqrt(self.K))  
    def predict(self,uid,iid):  #预测评分的函数  
        #setdefault的作用是当该用户或者物品未出现过时,新建它的bi,bu,qi,pu,并设置初始值为0  
        self.bi.setdefault(iid,0)  
        self.bu.setdefault(uid,0)  
        self.qi.setdefault(iid,np.zeros((self.K,1)))  
        self.pu.setdefault(uid,np.zeros((self.K,1)))  
        rating=self.avg+self.bi[iid]+self.bu[uid]+np.sum(self.qi[iid]*self.pu[uid]) #预测评分公式  
        #由于评分范围在1到5,所以当分数大于5或小于1时,返回5,1.  
        if rating>5:  
            rating=5  
        if rating<1:  
            rating=1  
        return rating  
      
    def train(self,steps=30,gamma=0.04,Lambda=0.15):    #训练函数,step为迭代次数。  
        print('train data size',self.mat.shape)  
        for step in range(steps):  
            print('step',step+1,'is running')  
            KK=np.random.permutation(self.mat.shape[0]) #随机梯度下降算法,kk为对矩阵进行随机洗牌  
            rmse=0.0;mae=0
            for i in range(self.mat.shape[0]):  
                j=KK[i]  
                uid=self.mat[j,0]  
                iid=self.mat[j,1]  
                rating=self.mat[j,2]  
                eui=rating-self.predict(uid, iid)  
                rmse+=eui**2  
                mae+=abs(eui)
                self.bu[uid]+=gamma*(eui-Lambda*self.bu[uid])    
                self.bi[iid]+=gamma*(eui-Lambda*self.bi[iid])  
                tmp=self.qi[iid]  
                self.qi[iid]+=gamma*(eui*self.pu[uid]-Lambda*self.qi[iid])  
                self.pu[uid]+=gamma*(eui*tmp-Lambda*self.pu[uid])  
            gamma=0.93*gamma  #gamma以0.93的学习率递减 
            print('rmse is {0:3f}, ase is {1:3f}'.format(np.sqrt(rmse/self.mat.shape[0]),mae/self.mat.shape[0]))  
      
    def test(self,test_data):   
          
        test_data=np.array(test_data)  
        print('test data size',test_data.shape)  
        rmse=0.0;mae=0  
        for i in range(test_data.shape[0]):  
            uid=test_data[i,0]  
            iid=test_data[i,1]  
            rating=test_data[i,2]  
            eui=rating-self.predict(uid, iid)  
            rmse+=eui**2  
            mae+=abs(eui)
        print('rmse is {0:3f}, ase is {1:3f}'.format(np.sqrt(rmse/self.mat.shape[0]),mae/self.mat.shape[0]))  
              
              
def getData(file_name): 
    """
    获取训练集和测试集的函数 
    """  
    data=[] 
    with open(os.path.expanduser(file_name)) as f:                          
        for line in f.readlines():  
            list=line.split('::')  
            data.append([int(i) for i in list[:3]])  
    random.shuffle(data)  
    train_data=data[:int(len(data)*7/10)]  
    test_data=data[int(len(data)*7/10):]  
    print('load data finished')  
    print('total data ',len(data))  
    return train_data,test_data  
           
if __name__=='__main__':  
    train_data,test_data=getData('D:/Downloads/ml-1m/ratings.dat')  
    a=SVD(train_data,30)    
    a.train()  
    a.test(test_data)  

测试结果

在训练集上
rmse is 0.869038, ase is 0.690794
在测试集上
rmse is 0.583027, ase is 0.303116

 SVD++算法:

SVD算法是指在SVD的基础上引入隐式反馈,使用用户的历史浏览数据、用户历史评分数据等作为新的参数。

这里的N(u)表示用户u行为记录(包括浏览的和评过分的商品集合),yj为隐藏的“评价了电影 j”反映出的个人喜好偏置。其他参数同SVD中的参数含义一致。

 利用随机梯度下降算法更新参数:

代码体现:

# -*- coding: utf-8 -*-
"""
Created on Sat Apr 21 14:44:25 2018

@author: fanchao3
"""
import numpy as np  
import random
import os

class SVDPP:  
    def __init__(self,mat,K=20):  
        self.mat=np.array(mat)  
        self.K=K  
        self.bi={}  
        self.bu={}  
        self.qi={}  
        self.pu={}  
        self.avg=np.mean(self.mat[:,2])  
        self.y={}  
        self.u_dict={}  
        for i in range(self.mat.shape[0]):  
              
            uid=self.mat[i,0]  
            iid=self.mat[i,1]  
            self.u_dict.setdefault(uid,[])  
            self.u_dict[uid].append(iid)  
            self.bi.setdefault(iid,0)  
            self.bu.setdefault(uid,0)  
            self.qi.setdefault(iid,np.random.random((self.K,1))/10*np.sqrt(self.K))  
            self.pu.setdefault(uid,np.random.random((self.K,1))/10*np.sqrt(self.K))  
            self.y.setdefault(iid,np.zeros((self.K,1))+.1)  
    def predict(self,uid,iid):  #预测评分的函数  
        #setdefault的作用是当该用户或者物品未出现过时,新建它的bi,bu,qi,pu及用户评价过的物品u_dict,并设置初始值为0  
        self.bi.setdefault(iid,0)  
        self.bu.setdefault(uid,0)  
        self.qi.setdefault(iid,np.zeros((self.K,1)))  
        self.pu.setdefault(uid,np.zeros((self.K,1)))  
        self.y.setdefault(uid,np.zeros((self.K,1)))  
        self.u_dict.setdefault(uid,[])  
        u_impl_prf,sqrt_Nu=self.getY(uid, iid)  
        rating=self.avg+self.bi[iid]+self.bu[uid]+np.sum(self.qi[iid]*(self.pu[uid]+u_impl_prf)) #预测评分公式  
        #由于评分范围在1到5,所以当分数大于5或小于1时,返回5,1.  
        if rating>5:  
            rating=5  
        if rating<1:  
            rating=1  
        return rating  
      
    #计算sqrt_Nu和∑yj  
    def getY(self,uid,iid):  
        Nu=self.u_dict[uid]  
        I_Nu=len(Nu)  
        sqrt_Nu=np.sqrt(I_Nu)  
        y_u=np.zeros((self.K,1))  
        if I_Nu==0:  
            u_impl_prf=y_u  
        else:  
            for i in Nu:  
                y_u+=self.y[i]  
            u_impl_prf = y_u / sqrt_Nu  
          
        return u_impl_prf,sqrt_Nu  
      
    def train(self,steps=30,gamma=0.04,Lambda=0.15):    #训练函数,step为迭代次数。  
        print('train data size',self.mat.shape)  
        for step in range(steps):  
            print('step',step+1,'is running')  
            KK=np.random.permutation(self.mat.shape[0]) #随机梯度下降算法,kk为对矩阵进行随机洗牌  
            rmse=0.0  
            for i in range(self.mat.shape[0]):  
                j=KK[i]  
                uid=self.mat[j,0]  
                iid=self.mat[j,1]  
                rating=self.mat[j,2]  
                predict=self.predict(uid, iid)  
                u_impl_prf,sqrt_Nu=self.getY(uid, iid)  
                eui=rating-predict  
                rmse+=eui**2  
                self.bu[uid]+=gamma*(eui-Lambda*self.bu[uid])    
                self.bi[iid]+=gamma*(eui-Lambda*self.bi[iid])  
                self.pu[uid]+=gamma*(eui*self.qi[iid]-Lambda*self.pu[uid])  
                self.qi[iid]+=gamma*(eui*(self.pu[uid]+u_impl_prf)-Lambda*self.qi[iid])  
                for j in self.u_dict[uid]:  
                    self.y[j]+=gamma*(eui*self.qi[j]/sqrt_Nu-Lambda*self.y[j])  
                                      
            gamma=0.93*gamma  
            print('rmse is',np.sqrt(rmse/self.mat.shape[0]))  
      
    def test(self,test_data):  #gamma以0.93的学习率递减  
          
        test_data=np.array(test_data)  
        print('test data size',test_data.shape)  
        rmse=0.0  
        for i in range(test_data.shape[0]):  
            uid=test_data[i,0]  
            iid=test_data[i,1]  
            rating=test_data[i,2]  
            eui=rating-self.predict(uid, iid)  
            rmse+=eui**2  
        print('rmse of test data is',np.sqrt(rmse/test_data.shape[0]))  
      
      
def getData(file_name): 
    """
    获取训练集和测试集的函数 
    """  
    data=[] 
    with open(os.path.expanduser(file_name)) as f:                          
        for line in f.readlines():  
            list=line.split('::')  
            data.append([int(i) for i in list[:3]])  
    random.shuffle(data)  
    train_data=data[:int(len(data)*7/10)]  
    test_data=data[int(len(data)*7/10):]  
    print('load data finished')  
    print('total data ',len(data))  
    return train_data,test_data  
      
if __name__=='__main__':  
    train_data,test_data=getData('D:/Downloads/ml-1m/ratings.dat')  
    a=SVDPP(train_data,30)    
    a.train()  
    a.test(test_data)  
# -*- coding: utf-8 -*-
"""
Created on Thu Apr 19 17:53:34 2018

@author: 
"""

import numpy as np  
import random  
import os 
  
class SVDPP:  
    def __init__(self,mat,K=20):  
        self.mat=np.array(mat)  
        self.K=K 
        self.avg=np.mean(self.mat[:,2]) 
        self.user_num = len(set(self.mat[:,0]))
        self.item_num = len(set(self.mat[:,1]))
        #print("item_num:",self.item_num )
        #user bias
        self.bu = np.zeros(self.user_num, np.double)

        #item bias
        self.bi = np.zeros(self.item_num, np.double)

        #user factor
        self.p = np.zeros((self.user_num, self.K), np.double) + .1

        #item factor
        self.q = np.zeros((self.item_num, self.K), np.double) + .1

        #item preference facotor
        self.y = np.zeros((self.item_num, self.K), np.double) + .1
        self.u_items={}
        for i in range(self.mat.shape[0]):
            uid=self.mat[i,0]
            iid=self.mat[i,1]
            if uid not in self.u_items.keys():                
                self.u_items[uid]=[iid]
            else:
                self.u_items[uid].append(iid)
                
    def train(self,steps=30,gamma=0.04,Lambda=0.15):    #训练函数,step为迭代次数。  
        #print('train data size',self.mat.shape) 
        
        for step in range(steps):  
            print('step',step+1,'is running')  
            KK=np.random.permutation(self.mat.shape[0]) #随机梯度下降算法,kk为对矩阵进行随机洗牌  
            rmse=0.0;mae=0
            for i in range(self.mat.shape[0]):  
                j=KK[i]  
                uid=self.mat[j,0]  
                iid=self.mat[j,1]  
                rating=self.mat[j,2]
                Nu=self.u_items[uid]
                I_Nu = len(Nu)
                sqrt_N_u = np.sqrt(I_Nu)
                #基于用户u点评的item集推测u的implicit偏好
                y_u = np.sum(self.y[Nu], axis=0)
                u_impl_prf = y_u / sqrt_N_u
                #预测值
                rp = self.avg + self.bu[uid] + self.bi[iid] + np.dot(self.q[iid], self.p[uid] + u_impl_prf)
                eui=rating- rp
                rmse+=eui**2  
                mae+=abs(eui)
                #sgd
                self.bu[uid] += gamma * (eui - Lambda * self.bu[uid])
                self.bi[iid] += gamma * (eui - Lambda * self.bi[iid])
                self.p[uid] += gamma * (eui * self.q[iid] - Lambda * self.p[uid])
                self.q[iid] += gamma * (eui * (self.p[uid] + u_impl_prf) - Lambda * self.q[iid])
                for j in Nu:
                    self.y[j] += gamma * (eui * self.q[j] / sqrt_N_u - Lambda * self.y[j])

            gamma=0.93*gamma  #gamma以0.93的学习率递减 
            print('rmse is {0:3f}, ase is {1:3f}'.format(np.sqrt(rmse/self.mat.shape[0]),mae/self.mat.shape[0]))  
      
    def test(self,test_data):   
          
        test_data=np.array(test_data)  
        print('test data size',test_data.shape)  
        rmse=0.0;mae=0  
        for i in range(test_data.shape[0]):  
            uid=test_data[i,0]  
            iid=test_data[i,1]  
            Nu=self.u_items[uid]
            I_Nu = len(Nu)
            sqrt_N_u = np.sqrt(I_Nu)
            y_u = np.sum(self.y[Nu], axis=0) / sqrt_N_u
            est = self.avg + self.bu[uid] + self.bi[iid] + np.dot(self.q[iid], self.p[uid] + y_u)
            rating=test_data[i,2]  
            eui=rating-est  
            rmse+=eui**2  
            mae+=abs(eui)
        print('rmse is {0:3f}, ase is {1:3f}'.format(np.sqrt(rmse/self.mat.shape[0]),mae/self.mat.shape[0]))  
              
              
def getData(file_name): 
    """
    获取训练集和测试集的函数 
    """  
    data=[] 
    with open(os.path.expanduser(file_name)) as f:                          
        for line in f.readlines():
            List=line.split('::')  
            data.append([int(i) for i in List[:3]])  

    random.shuffle(data)  
    train_data=data[:int(len(data)*7/10)]  
    test_data=data[int(len(data)*7/10):]  
    new_train_data=mapping(train_data)
    new_test_data=mapping(test_data)
    print('load data finished')  
    return new_train_data,new_test_data  
def mapping(data):
    """
    将原始的uid,iid映射为从0开始的编号
    """
    data=np.array(data)
    users=list(set(data[:,0]))
    u_dict={}
    for i in range(len(users)):
        u_dict[users[i]]=i
    items=list(set(data[:,1]))
    i_dict={}
    for j in range(len(items)):
        i_dict[items[j]]=j
    new_data=[]
    for l in data:
        uid=u_dict[l[0]]
        iid=i_dict[l[1]]
        r=l[2]
        new_data.append([uid,iid,r])
    return new_data
     
if __name__=='__main__':  
    train_data,test_data=getData('D:/Downloads/ml-1m/ratings.dat')  
    a=SVDPP(train_data,30)    
    a.train()  
    a.test(test_data)  
原文地址:https://www.cnblogs.com/chaofn/p/8877105.html