Method load_from_csv

python/dgl/data/csv_dataset_base.py:145–159

(
        meta: MetaNode, data_parser: Callable, base_dir=None, separator=","
    )

    @staticmethod
    def load_from_csv(
        meta: MetaNode, data_parser: Callable, base_dir=None, separator=","
    ):
        df = BaseData.read_csv(meta.file_name, base_dir, separator)
        node_ids = BaseData.pop_from_dataframe(df, meta.node_id_field)
        graph_ids = BaseData.pop_from_dataframe(df, meta.graph_id_field)
        if node_ids is None:
            raise DGLError(
                "Missing node id field [{}] in file [{}].".format(
                    meta.node_id_field, meta.file_name
                )
            )
        ntype = meta.ntype
        ndata = data_parser(df)
        return NodeData(node_ids, ndata, type=ntype, graph_id=graph_ids)

    @staticmethod
    def to_dict(node_data: List["NodeData"]) -> dict: