Source code for ggfm.datasets.CiteSeer

import os
import os.path
from typing import Callable, List, Optional

from ggfm.data import graph as gg
from ggfm.data import download_url, extract_zip


class CiteSeer:
    r"""
    The CiteSeer dataset is a widely used benchmark in graph neural network research, comprising 3,312 scientific
    publications in the field of computer science. These publications are categorized into six classes: Agents, AI,
    DB, IR, ML, and HCI. The dataset includes a citation network with 4,732 links, where each publication is
    represented by a 3,703-dimensional binary feature vector indicating the presence or absence of specific words in
    the document.
    """

    url = 'todo'

    def __init__(self, root: str = None, transform: Optional[Callable] = None,
                 pre_transform: Optional[Callable] = None, force_reload: bool = False):
        super().__init__(root, transform, pre_transform, force_reload=force_reload)
        self.data, self.slices = self.load_data(self.processed_paths[0])

    @property
    def raw_file_names(self) -> List[str]:
        pass

    @property
    def processed_file_names(self) -> str:
        return 'CiteSeer_pre_data.pt'

    def download(self):
        path = download_url(self.url, self.raw_dir)
        extract_zip(path, self.raw_dir)
        os.remove(path)

    def process(self):
        data = gg()

        pass
        # todo

    def __repr__(self) -> str:
        return f'{self.__class__.__name__}()'