torch_geometric.datasets.MalNetTiny

class MalNetTiny(root: str, split: Optional[str] = None, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None, force_reload: bool = False)

Bases: InMemoryDataset

来自“A Large-Scale Database for Graph Representation Learning”论文的MalNet Tiny数据集。 MalNetTiny包含了5种不同类型的5000个恶意和良性软件函数调用图。每个图最多包含5000个节点。

Parameters:

• root (str) – Root directory where the dataset should be saved.

• split (str, optional) – 如果为 "train"，则加载训练数据集。 如果为 "val"，则加载验证数据集。 如果为 "trainval"，则加载训练和验证数据集。 如果为 "test"，则加载测试数据集。 如果为 None，则加载整个数据集。 (默认值: None)

• transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before every access. (default: None)

• pre_transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before being saved to disk. (default: None)

• pre_filter (callable, optional) – A function that takes in an torch_geometric.data.Data object and returns a boolean value, indicating whether the data object should be included in the final dataset. (default: None)

• force_reload (bool, optional) – Whether to re-process the dataset. (default: False)