Beispiel #1
0
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)
Beispiel #2
0
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()