ggfm.datasets.IMDB¶
- class ggfm.datasets.IMDB(root: str | None = None, transform: Callable | None = None, pre_transform: Callable | None = None, force_reload: bool = False)[source]¶
Bases:
objectThe IMDb dataset is a comprehensive collection of movie-related information, including details about movies, TV series, podcasts, home videos, video games, and streaming content. It encompasses data such as cast and crew, plot summaries, trivia, ratings, and user and critic reviews. This dataset is widely used for applications in recommendation systems, sentiment analysis, and media analytics. A subset of the IMDb movie database is used in the “MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding” <https://arxiv.org/abs/2002.01680>_ paper. This subset focuses on heterogeneous graph structures involving various entities such as movies, actors, directors, genres, and more, facilitating research in graph neural networks and their applications to multimedia data.