import poi
from utils import Filename
from utils import setup_log
from utils import save_model
if __name__ == "__main__":
mdname = "bpr"
fn = Filename("foursquare")
setup_log(fn.log(mdname))
train_cks = poi.load_checkins(open(fn.train))
test_cks = poi.load_checkins(open(fn.test))
eva = poi.Evaluation(test_cks, full=False)
def hook(model):
eva.assess(model)
save_model(model, "./model/model_%s_%i.pkl" % (mdname, model.current))
mf = poi.BPR(train_cks,
learn_rate = 0.1,
reg_user=0.08,
reg_item=0.08,
)
mf.train(after=hook)
# -*- coding: utf-8 -*-
import poi
from utils import *
if __name__ == "__main__":
setup_log()
fn = Filename("foursquare")
locations = poi.load_locations(open(fn.locations))
cks = poi.load_checkins(open(fn.train))
pl = poi.PowerLaw(cks, locations)
pl.count()
pl.guass(max_x=10.0)
pl.plot("fq_pic.png")
# save pl model to file for reuse.
save_model(pl, "model/fq_powerlaw.pkl")
import poi
from utils import Filename
from utils import setup_log
from utils import save_model
if __name__ == "__main__":
mdname = "bpr"
fn = Filename("foursquare")
setup_log(fn.log(mdname))
train_cks = poi.load_checkins(open(fn.train))
test_cks = poi.load_checkins(open(fn.test))
eva = poi.Evaluation(test_cks, full=False)
def hook(model):
eva.assess(model)
save_model(model, "./model/model_%s_%i.pkl" % (mdname, model.current))
mf = poi.BPR(
train_cks,
learn_rate=0.1,
reg_user=0.08,
reg_item=0.08,
)
mf.train(after=hook)
