diff --git a/models/match/multiview-simnet/config.yaml b/models/match/multiview-simnet/config.yaml index 8399b40d4074f64d63e74e9cc0573ac6b422082c..53ac4c095c0d347cca8cba1afb9866c66ab85218 100755 --- a/models/match/multiview-simnet/config.yaml +++ b/models/match/multiview-simnet/config.yaml @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. evaluate: - workspace: "paddlerec.models.recall.multiview-simnet" + workspace: "paddlerec.models.match.multiview-simnet" reader: batch_size: 2 class: "{workspace}/evaluate_reader.py" @@ -24,7 +24,7 @@ train: strategy: "async" epochs: 2 - workspace: "paddlerec.models.recall.multiview-simnet" + workspace: "paddlerec.models.match.multiview-simnet" reader: batch_size: 2 diff --git a/models/match/readme.md b/models/match/readme.md index 1115eabd88461439e1d40c4ba808bdcd6d36b23c..6bccc109ff14582e816dee64b72b786a1e90f49e 100755 --- a/models/match/readme.md +++ b/models/match/readme.md @@ -9,13 +9,7 @@ * [整体介绍](#整体介绍) * [匹配模型列表](#匹配模型列表) * [使用教程](#使用教程) - * [数据处理](#数据处理) - * [训练](#训练) - * [预测](#预测) -* [效果对比](#效果对比) - * [模型效果列表](#模型效果列表) -* [分布式](#分布式) - * [模型性能列表](#模型性能列表) + * [训练&预测](#训练&预测) ## 整体介绍 ### 匹配模型列表 @@ -38,21 +32,9 @@

## 使用教程 -### 数据处理 -### 训练 -### 预测 - -## 效果对比 -### 模型效果列表 - -| 数据集 | 模型 | loss | auc | -| :------------------: | :--------------------: | :---------: |:---------: | -| - | DSSM | -- | -- | -| - | MultiView-Simnet | -- | -- | - -## 分布式 -### 模型性能列表 -| 数据集 | 模型 | 单机 | 多机（同步） | 多机（异步） | GPU | -| :------------------: | :--------------------: | :---------: |:---------: |:---------: |:---------: | -| - | DSSM | -- | -- | -- | -- | -| - | MultiView-Simnet | -- | -- | -- | -- | +### 训练&预测 +```shell +python -m paddlerec.run -m paddlerec.models.match.dssm # dssm +python -m paddlerec.run -m paddlerec.models.match.multiview-simnet # multiview-simnet +``` + diff --git a/models/multitask/readme.md b/models/multitask/readme.md index 3f31035351747ada8bcee3652f709c795a86e98b..d234f42f146e18bf254e518db0e78acc1e1d3e10 100755 --- a/models/multitask/readme.md +++ b/models/multitask/readme.md @@ -9,8 +9,7 @@ * [整体介绍](#整体介绍) * [多任务模型列表](#多任务模型列表) * [使用教程](#使用教程) - * [训练](#训练) - * [预测](#预测) + * [训练&预测](#训练&预测) * [效果对比](#效果对比) * [模型效果列表](#模型效果列表) @@ -42,16 +41,17 @@

## 使用教程 -### 训练 +### 训练&预测 ```shell -python -m paddlerec.run -m config.yaml -d cpu -e single +python -m paddlerec.run -m paddlerec.models.multitask.mmoe # mmoe +python -m paddlerec.run -m paddlerec.models.multitask.share-bottom # share-bottom +python -m paddlerec.run -m paddlerec.models.multitask.esmm # esmm ``` -### 预测 ## 效果对比 ### 模型效果列表 -| 数据集 | 模型 | loss | 评价指标 | +| 数据集 | 模型 | loss | auc | | :------------------: | :--------------------: | :---------: |:---------: | | Census-income Data | Share-Bottom | -- | 0.93120/0.99256 | | Census-income Data | MMoE | -- | 0.94465/0.99324 | diff --git a/models/rank/readme.md b/models/rank/readme.md index 326fb481356982dfb2acccaba670c072363bdb76..ac75c89fdd6b2d734aec8c8e8408f9036753d4ec 100755 --- a/models/rank/readme.md +++ b/models/rank/readme.md @@ -58,8 +58,15 @@ ## 使用教程 ### 数据处理 +参考每个模型目录数据下载&预处理脚本 ### 训练 +``` +python -m paddlerec.run -m paddlerec.models.rank.dnn # 以DNN为例 +``` ### 预测 +``` +python -m paddlerec.run -m paddlerec.models.rank.dnn # 以DNN为例 +``` ## 效果对比 ### 模型效果 (测试) @@ -69,7 +76,7 @@ | Criteo | DNN | -- | 0.79395 | -- | -- | | Criteo | DeepFM | 0.44797 | 0.80460 | -- | -- | | Criteo | DCN | 0.44704 | 0.80654 | -- | -- | -| Criteo | xDeepFM | -- | -- | 0.48657 | -- | +| Criteo | xDeepFM | 0.48657 | -- | -- | -- | | Census-income Data | Wide&Deep | 0.76195 | 0.90577 | -- | -- | | Amazon Product | DIN | 0.47005 | 0.86379 | -- | -- | diff --git a/models/recall/readme.md b/models/recall/readme.md index a86663e2da2bb99a3f6fc5c18a8b1800e77674a7..806e0a99e7576b96ee6d64bb1acc9e695dacb281 100755 --- a/models/recall/readme.md +++ b/models/recall/readme.md @@ -9,13 +9,9 @@ * [整体介绍](#整体介绍) * [召回模型列表](#召回模型列表) * [使用教程](#使用教程) - * [数据处理](#数据处理) - * [训练](#训练) - * [预测](#预测) + * [训练&预测](#训练&预测) * [效果对比](#效果对比) * [模型效果列表](#模型效果列表) -* [分布式](#分布式) - * [模型性能列表](#模型性能列表) ## 整体介绍 ### 召回模型列表 @@ -50,27 +46,20 @@

## 使用教程 -### 数据处理 - -### 训练 - -### 预测 - +### 训练&预测 +```shell +python -m paddlerec.run -m paddlerec.models.recall.word2vec # word2vec +python -m paddlerec.run -m paddlerec.models.recall.ssr # ssr +python -m paddlerec.run -m paddlerec.models.recall.gru4rec # gru4rec +python -m paddlerec.run -m paddlerec.models.recall.gnn # gnn +``` ## 效果对比 ### 模型效果列表 | 数据集 | 模型 | loss | Recall@20 | | :------------------: | :--------------------: | :---------: |:---------: | | DIGINETICA | GNN | -- | 0.507 | -| RSC15 | GRU4REC | -- | 0.67 | -| RSC15 | SSR | -- | 无 | +| RSC15 | GRU4REC | -- | 0.670 | +| RSC15 | SSR | -- | 0.590 | | 1 Billion Word Language Model Benchmark | Word2Vec | -- | 0.54 | -## 分布式 -### 模型性能列表 -| 数据集 | 模型 | 单机 | 多机（同步） | 多机（异步） | GPU | -| :------------------: | :--------------------: | :---------: |:---------: |:---------: |:---------: | -| DIGINETICA | GNN | -- | -- | -- | -- | -| RSC15 | GRU4REC | -- | -- | -- | -- | -| RSC15 | SSR | -- | -- | -- | -- | -| 1 Billion Word Language Model Benchmark | Word2Vec | -- | -- | -- | -- |