Graph4Rec模板文件执行出问题

[xinDongY]

运行如下代码时总是报错
node_types_file = "./rec_data/node_types.txt"
edges_file = "./rec_data/user2movie.txt"
embed_path = "./env_run/src/save_embed/embedding.txt"
movie_name_file = "./rec_data/nid2movieName.txt"

nid2type = load_vocab(node_types_file)
movie2vec, vec_dim = load_embed(embed_path, nid2type, dtype="movie")
user2vec, _ = load_embed(embed_path, nid2type, dtype="user")

print("building ann tree")
annIndex = AnnTree(user2vec, vec_dim)

print("loading user2likedMovies")
user2likedMovies = load_user2movies(edges_file)

print("loading movie2name")
movie2name = load_movie2name(movie_name_file)

---

报错如下

UnboundLocalError Traceback (most recent call last)
/tmp/ipykernel_199/1169641705.py in
8 nid2type = load_vocab(node_types_file)
9 movie2vec, vec_dim = load_embed(embed_path, nid2type, dtype="movie")
---> 10 user2vec, _ = load_embed(embed_path, nid2type, dtype="user")
11
12 print("building ann tree")

~/Graph4Rec/evaluation/evaluation.py in load_embed(embed_file, metaID2type, dtype)
48 embed = [float(i) for i in fields[1].split(" ")]
49 nodeID2vec[node_id] = embed
---> 50 vec_dim = len(embed)
51
52 return nodeID2vec, vec_dim

UnboundLocalError: local variable 'embed' referenced before assignment

不清楚为何内置函数的变量会被分配。感谢解答

[Liwb5]

你的embedding 文件是空的吧。

[Liwb5]

可能是因为你没有启动图引擎就开始训练，这样就没有节点被训练到，embedding文件就是空的。