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)