TabbedDataset

class TabbedDataset(cache_root: str | None = None, eager: bool = False, create_inverse_triples: bool = False, random_state: None | int | Generator = None)[source]

基础类：LazyDataset

这个类适用于当你有一个单独的TSV边缘文件并希望它们自动分割时。

初始化数据集。

Parameters:

  • cache_root (Path) – 一个可选的目录，用于存储提取的文件。如果没有给出，则使用默认的PyKEEN目录。 这是由环境变量PYKEEN_HOME定义的，或者默认为~/.pykeen。

  • eager (bool) – 数据是否应该立即加载？默认为 false。

  • create_inverse_triples (bool) – 是否应该创建反向三元组？默认为 false。

  • random_state (TorchRandomHint) – 一个可选的随机状态，用于使训练/测试/验证分割可重复。

属性摘要

ratios

属性文档

ratios: ClassVar[Sequence[float]] = (0.8, 0.1, 0.1)