import pycurl
from io import BytesIO
from pprint import pprint
import tornado.httpserver
import tornado.ioloop
import tornado.options
import tornado.web
from tornado.options import define, options
define("port", default=8000, help="run on the given port", type=int)
AppId = 'wxb9db0419592dbe7f'
AppSecret = '3a54d5663c95800880c927b124a31ead'
from util import Helper
helper = Helper(AppId, AppSecret)
Debug = True
def getKeyValueByURL(url, key):
buffers = BytesIO()
request_page = pycurl.Curl()
request_page.setopt(request_page.URL, url)
request_page.setopt(request_page.WRITEDATA, buffers)
try:
request_page.perform()
request_page.close()
jsonData = buffers.getvalue().decode()
return json.loads(jsonData)[key]
except pycurl.error, error:
mean_ndcgs.append(ndcg)
return mean_hrs, mean_ndcgs
if __name__ == '__main__':
args = parse_args()
args.lr = eval(args.lr)
args.path = './data/' + args.dataset + '/'
args.user_dataset = args.path + 'userRating'
args.group_dataset = args.path + 'groupRating'
args.user_in_group_path = "./data/" + args.dataset + "/groupMember.txt"
print('args', args)
# initial helper
helper = Helper()
# get the dict of users in group
g_m_d = helper.gen_group_member_dict(args.user_in_group_path)
# initial dataSet class
dataset = GDataset(args.user_dataset, args.group_dataset,
args.num_negatives, g_m_d)
# get group number
num_group = max(g_m_d.keys()) + 1
num_users, num_items = dataset.num_users, dataset.num_items
# print('num_users',num_users)
# build AGREE model
agree = AGREE(num_users, num_items, num_group, args.embedding_size, g_m_d,
args.drop_ratio).to(device)
