使用Movielens-20m创建DGL异质图
数据:
code
def add_MG():
""" 根据MovieID_genreID.csv创建MG边 """
# 当加载csv文件的多列数据时可以使用unpack将加载的数据列进场解耦到不同数组中
filename = "data/Movielens-20m/MovieID_genreID.csv"
movies, genres = np.loadtxt(filename, delimiter=",", skiprows=1, usecols=(0, 1),
原创
2021-07-13 17:44:05 ·
563 阅读 ·
0 评论