与上一篇UserCF数据来源一致。
先贴代码ItemCF:
#coding=utf-8
import math
class ItemCF:
def __init__(self,basefile,testfile):
self.datafile = basefile
self.testfile = testfile
self.readData()
self.readTestData()
def readData(self):
self.traindata = {}
for line in open(self.datafile):
userid,itemid,record,_ = line.split()
self.traindata.setdefault(userid,{})
self.traindata[userid][itemid] = int(record)
def readTestData(self):
self.testdata = {}
for line in open(self.testfile):
userid,itemid,record,_ = line.split()
self.testdata.setdefault(userid,{})
self.testdata[userid][itemid] =