RecBole小白入门系列博客(四)
——Sequential类模型运行流程
注意:本系列基于RecBole v.0.2.0 版本!
传送门:
RecBole小白入门系列博客(一)——快速安装和简单上手
RecBole小白入门系列博客(二)——General类模型运行流程
RecBole小白入门系列博客(三)——Context类模型运行流程
写在前面
- 这次的内容和General类和Context模型运行流程差不多,我会特别提一下Sequential类模型不一样和需要注意的地方。
- 由于模型的完整运行流程会牵扯到很多部分,从数据到参数,所以这里将简略的举出例子,让大家可以大致独立完成模型的运行。关于一些更加具体和细微的部分,我将在后面单独写博客为大家介绍。
选定模型
RecBole中共有17个Sequential类模型可供使用,模型列表如下:
- Sequential模型列表
关于模型的具体信息可以查看上面的链接,大家选用需要的模型即可。我们以模型GRU4Rec为例。
设置模型超参数
每个模型的超参数都不一致,大家可以从上面的链接中查看自己所需模型的超参数有哪些。在RecBole中每个模型都有默认的参数,因此如果是想简单跑一跑就暂时不动。
模型默认参数在源码目录RecBole/recbole/properties/model 中,各位感兴趣在小伙伴可以在里面查看默认参数值。
那么想调参的同学怎么办呢?读过RecBole小白入门系列博客(一)——快速安装和简单上手的朋友可能知道,在RecBole中调整参数有好几种方法,这里我推荐使用自定义配置文件。
RecBole的配置文件格式均使用.yaml文件,十分简洁明了,大致的形式如下:
accum: "stack"
gcn_output_dim: 500
embedding_size: 64
dropout_prob: 0.3
sparse_feature: True
class_num: 2
num_basis_functions: 2
参数名称: 参数值这样的样式
参数值可以是字典嵌套,也可以是列表等,就像这样
load_col:
inter: [user_id, business_id,
RecBole中Sequential类模型运行流程

本博客基于RecBole v.0.2.0版本,介绍Sequential类模型运行流程。包括选定模型,如GRU4Rec;设置模型超参数,推荐用自定义配置文件;选定数据集,介绍其基本格式;设置数据集、训练、评测参数;总结参数设置后运行模型,还提及调参及特殊模型设置问题。
最低0.47元/天 解锁文章
6532





