rdf2vecgpu.reader package

Submodules

rdf2vecgpu.reader.kg_file_reader module

class KGFileReader(file_path, multi_gpu, col_map=None, read_kwargs=None, walk_weighted=False)[source]

Bases: object

Parameters:
  • file_path (str)

  • multi_gpu (bool)

  • col_map (dict[str, str] | None)

  • read_kwargs (dict | None)

  • walk_weighted (bool)

read()[source]
Return type:

cudf.DataFrame | dask.dataframe.DataFrame

Module contents