UCIHousing

class paddle.text. UCIHousing ( data_file=None, mode='train', download=True ) [源代码] 

该类是对 UCI housing 测试数据集的实现。

参数

data_file (str) - 保存数据的路径，如果参数 download 设置为 True，可设置为 None。默认为 None。

mode (str) - 'train' 或 'test' 模式。默认为 'train'。

download (bool) - 如果 data_file 未设置，是否自动下载数据集。默认为 True。

返回值

Dataset，UCI housing 数据集实例。

代码示例

          >>> import paddle
>>> from paddle.text.datasets import UCIHousing

>>> class SimpleNet(paddle.nn.Layer):
...     def __init__(self):
...         super().__init__()
...
...     def forward(self, feature, target):
...         return paddle.sum(feature), target

>>> paddle.disable_static()

>>> uci_housing = UCIHousing(mode='train')

>>> for i in range(10):
...     feature, target = uci_housing[i]
...     feature = paddle.to_tensor(feature)
...     target = paddle.to_tensor(target)
...
...     model = SimpleNet()
...     feature, target = model(feature, target)
...     print(feature.shape, target.numpy())
paddle.Size([]) [24.]
paddle.Size([]) [21.6]
paddle.Size([]) [34.7]
paddle.Size([]) [33.4]
paddle.Size([]) [36.2]
paddle.Size([]) [28.7]
paddle.Size([]) [22.9]
paddle.Size([]) [27.1]
paddle.Size([]) [16.5]
paddle.Size([]) [18.9]

         