model MIND #398
Merged
model MIND #398
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
|
yinhaofeng
reviewed
Mar 19, 2021
models/recall/mind/README.md
Outdated
| 在mind模型目录的快速执行命令如下： | ||
| ``` | ||
| # 进入模型目录 | ||
| # cd models/recall/word2vec # 在任意目录均可运行 |
models/recall/mind/README.md
Outdated
| # 动态图训练 | ||
| python -u ../../../tools/trainer.py -m config.yaml | ||
| # 动态图预测 | ||
| python -u evaluate_dygraph.py -m config.yaml -top_n 50 #对测试数据进行预测，并通过faiss召回候选结果评测Reacll、NDCG、HitRate指标 |
There was a problem hiding this comment.
如果需要重写tools下的infer.py和static_infer.py，需要保持命名上的统一，名称还叫infer.py。可以参考word2vec
models/recall/mind/README.md
Outdated
| ``` | ||
| 3. 切回模型目录,执行命令运行全量数据 | ||
| ```bash | ||
| d - # 切回模型目录 |
models/recall/mind/README.md
Outdated
| │ ├── demo #demo训练数据 | ||
| │ │ └── demo.txt | ||
| │ ├── processs.py #处理全量数据的脚本 | ||
| │ ├── run.sh #全量数据下载的脚本 |
There was a problem hiding this comment.
可以将处理完成的全量数据加入BOS中，run.sh放在paddlerec/dataset目录下，方便用户直接使用以及切换数据集
models/recall/mind/net.py
Outdated
| mean=0.0, std=self.init_std)) | ||
| # paddle.static.Print(self.routing_logits) | ||
|
|
||
| # 仿射变换的参数 |
| @@ -0,0 +1,217 @@ | |||
| # Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. | |||
| @@ -0,0 +1,198 @@ | |||
| # Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. | |||
| @@ -0,0 +1,72 @@ | |||
| # Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. | |||
models/recall/mind/config.yaml
Outdated
| neg_samples: 1280 | ||
| maxlen: 20 | ||
| pow_p: 1.0 | ||
| distributed_embedding: 0 |
yinhaofeng
previously approved these changes
Mar 19, 2021
yinhaofeng
reviewed
Mar 19, 2021
models/recall/mind/README.md
Outdated
| # MIND(Multi-Interest Network with Dynamic Routing) | ||
|
|
||
| 以下是本例的简要目录结构及说明： | ||
| ```shell |
There was a problem hiding this comment.
readme中，可以直接拿出来执行的用bash，不能执行的直接用```即可
seemingwang
approved these changes
Mar 23, 2021
wangzhen38
pushed a commit
to wangzhen38/PaddleRec
that referenced
this pull request
Aug 6, 2021
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
实现MIND（Multi-Interest Network with Dynamic Routing forRecommendation at Tmall）
http://cn.arxiv.org/pdf/1904.08030v1