UCIHousing

class paddle.text.datasets.UCIHousing [源代码]

该类是对`UCI housing <https://archive.ics.uci.edu/ml/datasets/Housing>`_ 测试数据集的实现。

参数

  • data_file（str）- 保存数据的路径，如果参数:attr:`download`设置为True，

可设置为None。默认为None。 - mode（str）- ‘train’或’test’模式。默认为’train’。 - download（bool）- 如果:attr:`data_file`未设置，是否自动下载数据集。默认为True。

返回值

Dataset，UCI housing数据集实例。

代码示例

  1. import paddle
  2. from paddle.text.datasets import UCIHousing
  4. class SimpleNet(paddle.nn.Layer):
  5.     def __init__(self):
  6.         super(SimpleNet, self).__init__()
  8.     def forward(self, feature, target):
  9.         return paddle.sum(feature), target
  12. uci_housing = UCIHousing(mode='train')
  14. for i in range(10):
  15.     feature, target = uci_housing[i]
  16.     feature = paddle.to_tensor(feature)
  17.     target = paddle.to_tensor(target)
  19.     model = SimpleNet()
  20.     feature, target = model(feature, target)
  21.     print(feature.numpy().shape, target.numpy())