def simple(): from util import reader r = reader.stringSepReader("u.data", "\t") from util import split train, test1 = split.split(r.getR(), 1234567890) from recommender import nonpersonalized c = nonpersonalized.randomRec(r.getR(), 3284092) from util import test test.auc(test1, c.getRec, r)
def knn(): from util import reader r = reader.stringSepReader("u.data", "\t") from util import split train, test1 = split.splitMatrix(r.getMatrix(), 1234567890) from recommender import knn k = knn.userKnn(train, 10) from util import test print test.hitrate(test1, k.getRec, 10) import cPickle output = open("knn.npz", "wb") cPickle.dump(k, output, -1) output.close()