mindspore_gl.dataset.MetrLa

查看源文件
class mindspore_gl.dataset.MetrLa(root)[源代码]

METR-LA,用于读取和解析METR-LA数据集的源数据集。

关于METR-LA数据集:

METR-LA是一个大规模数据集,由1500个交通环路探测器在洛杉矶乡村道路网络中收集而来。该数据集包括速度、流量和占用率数据,涵盖了大约3420英里的道路。

数据:

  • 时间步: 12,6850

  • 节点: 207

  • 边: 1515

下载地址:METR-LA 。您可以将数据集文件组织到以下目录结构中,并通过 mindspore_gl.dataset.MetrLa.get_data API读取。

.
├── adj_mat.npy
└── node_values.npy
参数:
  • root (str) - 包含METR-LA/adj_mat.npy和METR-LA/node_values.npy的根目录路径。

输入:
  • in_timestep (int) - 输入时序数。

  • out_timestep (int) - 输出时序数。

异常:
  • TypeError - 如果 root 不是str。

  • RuntimeError - 如果 root 不包含数据文件。

  • TypeError - 如果 in_timestepout_timestep 不是正整数。

样例:

>>> from mindspore_gl.dataset.ppi import MetrLa
>>> root = "path/to/metrla"
>>> dataset = MetrLa(root)
>>> features, labels = dataset.get_data(in_timestep, out_timestep)
get_data(in_timestep, out_timestep)[源代码]

获取序列时间特征和标签。

参数:
  • in_timestep (int) - 输入时序数。

  • out_timestep (int) - 输出时序数。

property node_count

节点数。

返回:

int,节点数。

样例:

>>> #dataset is an instance object of Dataset
>>> node_count = dataset.node_count