Movielens

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

该类是对`Movielens 1-M <https://grouplens.org/datasets/movielens/1m/>`_ 测试数据集的实现。

参数

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

可设置为None。默认为None。 - mode（str）- ‘train’ 或 ‘test’ 模式。默认为’train’。 - test_ratio（float) - 为测试集划分的比例。默认为0.1。 - rand_seed（int）- 随机数种子。默认为0。 - download（bool）- 如果:attr:`data_file`未设置，是否自动下载数据集。默认为True。

返回值

Dataset，Movielens 1-M数据集实例。

代码示例

  1. import paddle
  2. from paddle.text.datasets import Movielens
  4. class SimpleNet(paddle.nn.Layer):
  5.     def __init__(self):
  6.         super(SimpleNet, self).__init__()
  8.     def forward(self, category, title, rating):
  9.         return paddle.sum(category), paddle.sum(title), paddle.sum(rating)
  12. movielens = Movielens(mode='train')
  14. for i in range(10):
  15.     category, title, rating = movielens[i][-3:]
  16.     category = paddle.to_tensor(category)
  17.     title = paddle.to_tensor(title)
  18.     rating = paddle.to_tensor(rating)
  20.     model = SimpleNet()
  21.     category, title, rating = model(category, title, rating)
  22.     print(category.numpy().shape, title.numpy().shape, rating.numpy().shape)