Ejemplo n.º 1
0
    user=1                          #gio y cho user 1
    loaiDadang=0;
    alpha=0.5                       #can bang da dang voi rel
    standardMetric=0.8              #
    kUser=5                         #k_nearest_neighbors 
#   đường dẫn    
    baseFile = 'D:/danthanh/DoAnTotNghiep/datasets/ml-100k/ua.base'
    itemFile = 'D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u.item'
    userFile='D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u.user'
#   
#    #khoi tao listitem, va ma tran danh gia
    listItem=ReadFile.getItem(itemFile);
    listUser=ReadFile.getUser(userFile);
    urm=ReadFile.UserRatingMatrix("D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u1.base",MAX_UID,MAX_IID)
    urmTest=ReadFile.UserRatingMatrix("D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u1.test",MAX_UID,MAX_IID)
    uTest=ReadFile.GetListUSerTest("D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u1.test")
    gMS=ReadFile.getMoviesSeen("D:/danthanh/DoAnTotNghiep/datasets/ml-100k/u1.base")
    GroupByCategory,GroupByMovies=ReadFile.getMovies_Genres(itemFile)


############################### MF   #####################################
    #    Hệ thống cũ MF
    U, S, Vt = Function.computeSVD(urm, K)  
    A=U*S*Vt
    listRecommedOld = Function.MF(U, S, Vt, uTest, gMS,topN,MAX_UID,MAX_IID)
#    fullRecommedMF=listRecommedOld
#    print (fullRecommedMF)
    userRecommedMF=listRecommedOld[user];
    print ('Danh sach goi y cu bang MF:')
    print('################################################');
    print (userRecommedMF)