diff --git a/core/trainers/transpiler_trainer.py b/core/trainers/transpiler_trainer.py index 3dc8bfd6e04fb3da2b217f84126fac8101102945..2aed5bfbedfd264b89671ba0fa776ff716042cf5 100755 --- a/core/trainers/transpiler_trainer.py +++ b/core/trainers/transpiler_trainer.py @@ -42,7 +42,7 @@ class TranspileTrainer(Trainer): namespace = "train.reader" class_name = "TrainReader" else: - readerdataloader = self.model._infer_data_loader + dataloader = self.model._infer_data_loader namespace = "evaluate.reader" class_name = "EvaluateReader" diff --git a/readme.md b/readme.md index 4c63fc1210cedc3e4c70a21a48f49fed453872a0..25020fcf73df5a1a70853a14e65c57d1d5076d55 100644 --- a/readme.md +++ b/readme.md @@ -114,32 +114,40 @@ python -m paddlerec.run -m ./models/rank/dnn/config.yaml -e single | 内容理解 | [Text-Classifcation](models/contentunderstanding/text_classification/model.py) | ✓ | x | ✓ | x | | 内容理解 | [TagSpace](models/contentunderstanding/tagspace/model.py) | ✓ | x | ✓ | x | | 召回 | [Word2Vec](models/recall/word2vec/model.py) | ✓ | x | ✓ | x | -| 召回 | [TDM](models/recall/tdm/model.py) | ✓ | x | ✓ | x | +| 召回 | [TreeBased-Model](models/recall/tdm/model.py) | ✓ | x | ✓ | x | | 召回 | [SSR](models/recall/ssr/model.py) | ✓ | ✓ | ✓ | x | | 召回 | [Gru4Rec](models/recall/gru4rec/model.py) | ✓ | ✓ | ✓ | x | -| 排序 | [CTR-Dnn](models/rank/dnn/model.py) | ✓ | x | ✓ | x | -| 排序 | [DeepFm](models/rank/deepfm/model.py) | ✓ | x | ✓ | x | -| 排序 | [xDeepFm](models/rank/xdeepfm/model.py) | ✓ | x | ✓ | x | +| 排序 | [Dnn](models/rank/dnn/model.py) | ✓ | x | ✓ | x | +| 排序 | [DeepFM](models/rank/deepfm/model.py) | ✓ | x | ✓ | x | +| 排序 | [xDeepFM](models/rank/xdeepfm/model.py) | ✓ | x | ✓ | x | | 排序 | [DIN](models/rank/din/model.py) | ✓ | x | ✓ | x | | 排序 | [Wide&Deep](models/rank/wide_deep/model.py) | ✓ | x | ✓ | x | -| 多任务 | [ESMM](models/multitask/essm/model.py) | ✓ | ✓ | ✓ | x | +| 多任务 | [ESMM](models/multitask/esmm/model.py) | ✓ | ✓ | ✓ | x | | 多任务 | [MMOE](models/multitask/mmoe/model.py) | ✓ | ✓ | ✓ | x | -| 排序 | [ShareBottom](models/multitask/share-bottom/model.py) | ✓ | ✓ | ✓ | x | +| 多任务 | [ShareBottom](models/multitask/share-bottom/model.py) | ✓ | ✓ | ✓ | x | | 匹配 | [DSSM](models/match/dssm/model.py) | ✓ | x | ✓ | x | -| 匹配 | [Simnet](models/match/multiview-simnet/model.py) | ✓ | x | ✓ | x | +| 匹配 | [MultiView-Simnet](models/match/multiview-simnet/model.py) | ✓ | x | ✓ | x |