def get_neibor_embedding(fname):
data = thread_load(fname)
embedding = dict()
for uid, e in data.items():
if len(e) < 6:
continue
# embedding[uid] = list(e[0]) + list(e[1]) + list(e[2]) + list(e[3]) + list(e[4]) + list(e[5])
embedding[uid] = list(e[0]) + list(e[1]) + list(e[2][-1]) + list(e[3][
-1]) + list(e[4][-1]) + list(e[5][-1])
# embedding[uid] = list(e[0]) + list(e[1]) + list(e[3]) + list(e[5])
# embedding[uid]=e[0]+e[1]
return embedding
def get_neibor_embedding(fname, index):
data = thread_load(fname)
embedding = dict()
for uid, e in data.items():
if len(e) < 6:
continue
if e[2][index] == [] or e[3][index] == [] or e[4][index] == [] or e[5][
index] == []:
continue
embedding[uid] = list(e[2][index]) + list(e[3][index]) + list(e[4][
index]) + list(e[5][index])
return embedding
def get_simple_embedding(fname):
return thread_load(fname)
