ml_regression_cn.rst 12.6 KB
Newer Older
1 2 3
MovieLens数据集评分回归模型
=========================

D
dayhaha 已提交
4
这里我们在MovieLens数据集描述一种 **余弦相似度回归** 任务。
5 6 7 8 9 10 11 12 13 14 15 16 17
该示例将展示paddle如何进行词向量嵌入,处理相似度回归,针对文本
的单词级别的卷积神经网络,以及paddle如何处理多种类型的输入。
需要注意的是,该模型网络只是用于进行demo展示paddle如何工作,而
没有进行结构的微调。


**我们非常欢迎您用PADDLEPADDLE构建更好的示例,如果您有好的建议来
让这个示例变得更好,希望能让我们知晓。**

数据准备
```````
下载并解压数据集
''''''''''''''
D
dayhaha 已提交
18
这里我们使用 :ref:`demo_ml_dataset` 。
19 20 21 22 23 24 25
要下载和解压数据集,只需要简单的运行下面的命令即可。

.. code-block:: bash

	cd demo/recommendation/data
	./ml_data.sh

D
dayhaha 已提交
26
:code:`demo/recommendation/data/ml-1m` 的目录结构为:
27 28 29 30 31 32 33 34 35 36 37

.. code-block:: text

	+--ml-1m
		+--- movies.dat 	# 电影特征
		+--- ratings.dat 	# 评分
		+--- users.dat 		# 用户特征
		+--- README 		# 数据集描述

字段配置文件
''''''''''
D
dayhaha 已提交
38 39
**字段配置文件** 用来具体说明数据集的字段和文件格式,
例如,说明每个特征文件具体字段是 **什么** 类型。
40

D
dayhaha 已提交
41
ml-1m的字段配置文件在目录 :code:`demo/recommendation/data/config.json` 中。
42
其具体说明了字段类型和文件名称:
D
dayhaha 已提交
43

44
1) 用户文件中有四种类型的字段\: 编号,性别,年龄和职业;
D
dayhaha 已提交
45

46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
2) 文件名称为"users.dat",文件的分隔符为"::"。

.. include:: ../../../demo/recommendation/data/config.json
   :code: json
   :literal:

准备数据
```````
你需要安装python的第三方库。
**强烈推荐使用VIRTUALENV来创造一个干净的python环境。**

.. code-block:: bash

	pip install -r requirements.txt

预处理数据一般的命令为:

.. code-block:: bash

	cd demo/recommendation
	./preprocess.sh

下面介绍预处理过程具体的步骤。

提取电影或用户的特征并生成python对象
''''''''''''''''''''''''''''''''

在movielens 1m数据集中,电影和用户有许多的特征。
评分文件的每一行仅仅提供电影或用户的编号来代表相应的电影或用户。
D
dayhaha 已提交
75
我们首先处理电影或用户的特征文件,然后用pickle命令将特征( **Meta** )对象存储为文件。
76 77 78 79

Meta配置文件
...........

D
dayhaha 已提交
80
**Meta配置文件** 用来具体描述 **如何** 解析数据集中的每一个字段。
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
该文件可以从字段配置文件生成,或是手动编辑生成。文件的格式可以
为json或yaml格式。解析器能通过文件的扩展名自动识别文件的格式。

要将字段配置文件转化为meta配置文件,只需要运行:

.. code-block:: bash

	cd demo/recommendation/data
	python config_generator.py config.json > meta_config.json

生成的meta配置文件如下所示:

.. include:: ../../../demo/recommendation/data/meta_config.json
	:code: json
	:literal:

在meta文件中有两种特征\: 电影和用户。

* 在电影文件movies.dat中
	* 我们仅用"::"来分隔每一行
D
dayhaha 已提交
101
	* pos 0 代表编号
102
	* pos 1 特征:
D
dayhaha 已提交
103 104 105 106
		* name是电影名
		* 利用正则表达式来解析该特征
		* 基于字母的词嵌入特征
		* 是序列
107
	* pos 2 特征:
D
dayhaha 已提交
108 109 110
		* name是体裁
		* type是one hot稠密向量
		* dictionary由解析自动生成,每一个key由'|'分隔
111 112
* 在用户文件users.dat中
	* 我们仅用"::"来分隔每一行
D
dayhaha 已提交
113
	* pos 0 代表编号
114
	* pos 1 特征:
D
dayhaha 已提交
115 116
		* name是性别
		* 简单的基于字母的词嵌入
117 118 119 120 121 122 123 124 125 126 127 128
	* pos 2 特征:
		* name是年龄
		* 是整个的词嵌入
		* 嵌入编号会根据单词排序
	* pos 3 特征:
		* name是职业
		* 简单的整个词嵌入


Meta文件
''''''''

D
dayhaha 已提交
129
有了meta配置文件之后,我们可以生成 **Meta文件** ,该文件是python的pickle对象,
130 131 132 133 134 135
存储着电影或用户信息。可以运行下面的命令来生成。

.. code-block:: bash

	python meta_generator.py ml-1m meta.bin --config=meta_config.json

D
dayhaha 已提交
136
meta文件 :code:`meta.bin` 的结构如下:
137 138 139

.. code-block:: text

D
dayhaha 已提交
140
    +--+ movie
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
    |      +--+ __meta__
    |      |       +--+ raw_meta  # 每个特征的meta配置。列表
    |      |       |       +
    |      |       |       |     # 编号字段,我们用编号作为key 
    |      |       |       +--+ {'count': 3883, 'max': 3952, 'is_key': True, 'type': 'id', 'min': 1}
    |      |       |       |
    |      |       |       |     # 电影名字段,嵌入特征字典
    |      |       |       +--+ {'dict': [ ... ], 'type': 'embedding', 'name': 'title', 'seq': 'sequence'}
    |      |       |       |
    |      |       |       |     # 体裁字段,体裁字典
    |      |       |       +--+ {'dict': [ ... ], 'type': 'one_hot_dense', 'name': 'genres'}
    |      |       |
    |      |       +--+ feature_map [1, 2] # a list for raw_meta index for feature field.
    |      |                               # it means there are 2 features for each key.
    |      |                               #    * 0 offset of feature is raw_meta[1], Title.
    |      |                               #    * 1 offset of feature is raw_meta[2], Genres.
    |      |
    |      +--+ 1 # 电影1的特征
    |      |    +
    |      |    +---+ [[...], [...]] # title ids, genres dense vector
    |      |
    |      +--+ 2
    |      |
    |      +--+ ...
    |
    +--- user
           +--+ __meta__
           |       +
           |       +--+ raw_meta
           |       |       +
           |       |       +--+ id field as user
           |       |       |
           |       |       +--+ {'dict': ['F', 'M'], 'type': 'embedding', 'name': 'gender', 'seq': 'no_sequence'}
           |       |       |
           |       |       +--+ {'dict': ['1', '18', '25', '35', '45', '50', '56'], 'type': 'embedding', 'name': 'age', 'seq': 'no_sequence'}
           |       |       |
           |       |       +--+ {'dict': [...], 'type': 'embedding', 'name': 'occupation', 'seq': 'no_sequence'}
           |       |
           |       +--+ feature_map [1, 2, 3]
           |
           +--+ 1 # 用户1的特征
           |
           +--+ 2
           +--+ ...


分割训练/测试文件
'''''''''''''''

D
dayhaha 已提交
190
我们将 :code:`ml-1m/ratings.dat` 文件分割为训练和测试文件。分割文件的方法是:对于每位用户,我们将评分分成两部分。
191 192
这样的话每位用户在测试文件中将与训练文件含有同样的信息。

D
dayhaha 已提交
193
用 :code:`separate.py` 来分离训练和测试文件。
194 195 196 197 198

.. code-block:: bash

	python split.py ml-1m/ratings.dat --delimiter="::" --test_ratio=0.1

D
dayhaha 已提交
199 200
这样就会生成两个文件::code:`ml-1m/ratings.dat.train` 和 :code:`ml-1m/ratings.data.test` 。
将他们移动到目录 :code:`data` ,然后进行随机打乱,再为paddle的训练过程提供文件列表。
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221

..  code-block:: bash

    shuf ml-1m/ratings.dat.train > ratings.dat.train
    cp ml-1m/ratings.dat.test .
    echo "./data/ratings.dat.train" > train.list
    echo "./data/ratings.dat.test" > test.list


神经网络结构配置
``````````````

训练器配置文件
''''''''''''

网络结构如下图所示:

..  image:: rec_regression_network.png
    :align: center
    :alt: rec_regression_network

D
dayhaha 已提交
222
该示例的神经网络配置文件 :code:`trainer_config.py` 如下所示:
223 224 225 226 227

..  literalinclude:: ../../../demo/recommendation/trainer_config.py
    :language: python
    :lines: 15-

D
dayhaha 已提交
228
在文件 :code:`trainer_config.py` 中,我们仅仅是将每个特征种类映射到一个特征向量中,以下
229 230
展示了如何将每个特征映射到一个向量。

D
dayhaha 已提交
231 232
* :code:`id` \: 仅仅是简单的嵌入,然后添加一个全连接层。
* :code:`embedding` \:
233
    - 如果是序列,则先做嵌入,然后再做一次文本卷积网络操作,
D
dayhaha 已提交
234
      然后得到平均采样的结果。
235
    - 如果不是序列,则先做嵌入,然后添加一个全连接层。
D
dayhaha 已提交
236
* :code:`one_host_dense` \:
237 238
    - 仅仅是两个全连接层。

D
dayhaha 已提交
239
然后我们利用多输入的:code:`fc_layer` 全连接层将电影的每个特征结合成一个电影特征,
240 241
并且对用户的特征做同样的操作,也得到一个用户特征。然后我们求这两个特征的余弦相似度。

D
dayhaha 已提交
242
在这些网络中,我们用以下的一些:ref:`api_trainer_config` 中的接口。
243 244 245 246 247 248 249 250

*  数据层, :ref:`api_trainer_config_helpers_layers_data_layer`
*  全连接层, :ref:`api_trainer_config_helpers_layers_fc_layer`
*  嵌入层, :ref:`api_trainer_config_helpers_layers_embedding_layer`
*  文本投影层, :ref:`api_trainer_config_helpers_layers_context_projection`
*  采样层, :ref:`api_trainer_config_helpers_layers_pooling_layer`
*  余弦相似度层, :ref:`api_trainer_config_helpers_layers_cos_sim`
*  文本卷积采样层, :ref:`api_trainer_config_helpers_network_text_conv_pool`
D
dayhaha 已提交
251
*  声明Python数据源, :ref:`api_trainer_config_helpers_data_sources` 
252 253 254 255 256 257 258 259 260

数据提供脚本
'''''''''''

..  literalinclude:: ../../../demo/recommendation/dataprovider.py
    :language: python
    :lines: 15-

数据提供脚本仅仅是读取meta.bin和评分文件,生成训练需要的样本。
D
dayhaha 已提交
261
在脚本 :code:`dataprovider.py` 中,我们需要设置:
262 263

* obj.slots\: 特征的类型和维度。
D
dayhaha 已提交
264
* use_seq\: :code:`dataprovider.py` 中的数据是否为序列模式。
D
dayhaha 已提交
265
* process\: 返回数据的每一条样本给 :code:`paddle` 。
266

D
dayhaha 已提交
267
数据提供脚本的细节文档可以参考 :ref:`api_pydataprovider` 。
268 269 270 271 272 273

训练
````

准备好数据,配置了网络,编写好数据提供脚本后,现在我们可以开始paddle训练了。

D
dayhaha 已提交
274
代码 :code:`run.sh` 如下:
275 276 277 278 279

..  literalinclude:: ../../../demo/recommendation/run.sh
    :language: bash
    :lines: 16-

D
dayhaha 已提交
280
该脚本仅仅是开始一个paddle训练过程,将日志写入文件 :code:`log.txt` ,然后
281 282
打印在屏幕上。

D
dayhaha 已提交
283
脚本 :code:`run.sh` 中的每一行命令,请参考页面 :ref:`cmd_line_index` 。
284 285 286
这些参数的简短介绍如下:

*  config\: 告诉paddle哪个文件是神经网络的配置文件。
D
dayhaha 已提交
287
*  save_dir\: 告诉paddle将模型保存在: code:`./output` 中。
288 289 290
*  use_gpu\: 是否使用GPU,默认为不使用。
*  trainer_count\: 一台机器上面的线程数量。
*  test_all_data_in_one_period\: 每一个测试周期测试一次所有数据。否则,
D
dayhaha 已提交
291
   每个测试周期测试: code:`batch_size` 批次的数据。
D
dayhaha 已提交
292 293 294
*  log_period\: 在训练了: code:`log_period` 批次后打印日志。
*  dot_period\: 在每训练: code:`dot_period` 个批次后打印一个 :code:`.` 。
*  num_passes\: 训练至多: code:`num_passes` 轮。
295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315

如果训练过程启动成功的话,输出应该类似如下:

..  code-block:: text

    I0601 08:07:22.832059 10549 TrainerInternal.cpp:157]  Batch=100 samples=160000 AvgCost=4.13494 CurrentCost=4.13494 Eval:  CurrentEval:

    I0601 08:07:50.672627 10549 TrainerInternal.cpp:157]  Batch=200 samples=320000 AvgCost=3.80957 CurrentCost=3.48421 Eval:  CurrentEval:

    I0601 08:08:18.877369 10549 TrainerInternal.cpp:157]  Batch=300 samples=480000 AvgCost=3.68145 CurrentCost=3.42519 Eval:  CurrentEval:

    I0601 08:08:46.863963 10549 TrainerInternal.cpp:157]  Batch=400 samples=640000 AvgCost=3.6007 CurrentCost=3.35847 Eval:  CurrentEval:

    I0601 08:09:15.413025 10549 TrainerInternal.cpp:157]  Batch=500 samples=800000 AvgCost=3.54811 CurrentCost=3.33773 Eval:  CurrentEval:
    I0601 08:09:36.058670 10549 TrainerInternal.cpp:181]  Pass=0 Batch=565 samples=902826 AvgCost=3.52368 Eval:
    I0601 08:09:46.215489 10549 Tester.cpp:101]  Test samples=97383 cost=3.32155 Eval:
    I0601 08:09:46.215966 10549 GradientMachine.cpp:132] Saving parameters to ./output/model/pass-00000
    I0601 08:09:46.233397 10549 ParamUtil.cpp:99] save dir ./output/model/pass-00000
    I0601 08:09:46.233438 10549 Util.cpp:209] copy trainer_config.py to ./output/model/pass-00000
    I0601 08:09:46.233541 10549 ParamUtil.cpp:147] fileName trainer_config.py

D
dayhaha 已提交
316
模型被保存在 :code:`output/` 目录中。你可以在任何时候用 :code:`Ctrl-C` 来停止训练。
317 318 319 320 321 322 323 324 325 326

模型评估和预测
````````````

在训练了几个轮次以后,你可以对模型进行评估,得到最好轮次下的模型。运行下面命令即可:

.. code-block:: bash

    ./evaluate.sh 

D
dayhaha 已提交
327
你将看到如下的信息:
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348

.. code-block:: text

    Best pass is 00009,  error is 3.06949, which means predict get error as 0.875998002281
    evaluating from pass output/pass-00009

然后,你可以预测任何用户对于任何一部电影的评价,运行下面命令即可:

..  code-block:: bash

    python prediction.py 'output/pass-00009/'

预测程序将读取用户的输入,然后输出预测分数。用户预测的命令行界面如下:

..  code-block:: text

    Input movie_id: 9
    Input user_id: 4
    Prediction Score is 2.56
    Input movie_id: 8
    Input user_id: 2
D
dayhaha 已提交
349
    Prediction Score is 3.13