Views
No views yet
pytorch_model.binmodel.pyutils.pyargs.json1import torch, json
2from types import SimpleNamespace
3from model import SASRec
4from utils import build_index, data_partition
5
6args = SimpleNamespace(**json.load(open("args.json")))
7u2i_index, i2u_index = build_index(args.dataset)
8dataset = data_partition(args.dataset)
9usernum, itemnum = dataset[3], dataset[4]
10
11model = SASRec(usernum, itemnum, args)
12model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
13model.eval()