search

YuxiangRen / Heterogeneous-Deep-Graph-Infomax

HDGI code
59 stars 14 forks source link

IMDB extract Edge #6

Closed NlpRookie closed 4 years ago

NlpRookie commented 4 years ago

Hi, I try to run extractEdge, an error occurred

sp.coo_matrix ValueError: column index exceeds matrix dimensions

I download this movie dataset from https://www.kaggle.com/carolzhangdc/imdb-5000-movie-dataset

YuxiangRen commented 4 years ago

I didn't use the dataset from Kaggle. I add the original data "movie_metadata_3class.csv" to the data fold, you can check it.

If you have other questions, just let me know.