Graph4Rec的例子报错了

[xinDongY]

运行下面这段代码的时候报错了
node_types_file = "./rec_data/node_types.txt"
edges_file = "./rec_data/user2movie.txt"
embed_path = "./outputs_custom/metapath2vec.0721/embedding/"
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_817/749958165.py in
5
6 nid2type = load_vocab(node_types_file)
----> 7 movie2vec, vec_dim = load_embed(embed_path, nid2type, dtype="movie")
8 user2vec, _ = load_embed(embed_path, nid2type, dtype="user")
9

~/Graph4Rec/evaluation/evaluation.py in load_embed(embed_file, metaID2type, dtype)
50 vec_dim = len(embed)
51
---> 52 return nodeID2vec, vec_dim
53
54 def filter_nodes(nodeID2vec, target_nodeID_list):

UnboundLocalError: local variable 'vec_dim' referenced before assignment

不清楚 'vec_dim'在之前如何被使用过