ICEWS18数据集

class dgl.data.ICEWS18Dataset(mode='train', raw_dir=None, force_reload=False, verbose=False, transform=None)[source]

Bases: DGLBuiltinDataset

用于时间图的ICEWS18数据集

综合危机早期预警系统 (ICEWS18)

事件数据由社会政治行为者之间的编码互动组成（即个人、团体、部门和民族国家之间的合作或敌对行动）。该数据集包含从2018年1月1日到2018年10月31日的事件（24小时时间粒度）。

参考：

统计：

  • 训练样本：240

  • 有效示例：30

  • 测试示例：34

  • 每个图的节点数：23033

Parameters:

  • mode (str) – 加载训练/验证/测试数据。必须是 [‘train’, ‘valid’, ‘test’] 中的一个。

  • raw_dir (str) – Raw file directory to download/contains the input data directory. Default: ~/.dgl/

  • force_reload (bool) – Whether to reload the dataset. Default: False

  • verbose (bool) – Whether to print out progress information. Default: True.

  • transform (callable, optional) – A transform that takes in a DGLGraph object and returns a transformed version. The DGLGraph object will be transformed before every access.

is_temporal

数据集是否包含时间图

Type:

bool

示例

>>> # get train, valid, test set
>>> train_data = ICEWS18Dataset()
>>> valid_data = ICEWS18Dataset(mode='valid')
>>> test_data = ICEWS18Dataset(mode='test')
>>>
>>> train_size = len(train_data)
>>> for g in train_data:
....    e_feat = g.edata['rel_type']
....    # your code here
....
>>>
__getitem__(idx)[source]

通过索引获取图表

Parameters:

idx (int) – Item index

Returns:

The graph contains:

  • edata['rel_type']: edge type

Return type:

dgl.DGLGraph

__len__()[source]

数据集中的图表数量。

Return type:

int