shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.
https://deepctr-torch.readthedocs.io/en/latest/index.html
Apache License 2.0
2.95k stars 696 forks source link

新增多任务模型 Dev zsx mtl1 #255

Closed zanshuxun closed 2 years ago

zanshuxun commented 2 years ago
  1. Add multi-task models: SharedBottom, ESMM, MMOE, PLE
  2. Bugfix: https://github.com/shenweichen/DeepCTR-Torch/issues/240 https://github.com/shenweichen/DeepCTR-Torch/issues/232
codecov[bot] commented 2 years ago

Codecov Report

Merging #255 (90cae06) into master (2cd84f3) will increase coverage by 0.26%. The diff coverage is 93.23%.

@@            Coverage Diff             @@
##           master     #255      +/-   ##
==========================================
+ Coverage   92.31%   92.58%   +0.26%     
==========================================
  Files          30       35       +5     
  Lines        2069     2333     +264     
==========================================
+ Hits         1910     2160     +250     
- Misses        159      173      +14     
Flag Coverage Δ
pytest 92.58% <93.23%> (+0.26%) :arrow_up:

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
deepctr_torch/models/basemodel.py 86.45% <87.50%> (+1.35%) :arrow_up:
deepctr_torch/models/multitask/esmm.py 90.24% <90.24%> (ø)
deepctr_torch/models/multitask/sharedbottom.py 91.30% <91.30%> (ø)
deepctr_torch/models/multitask/mmoe.py 92.53% <92.53%> (ø)
deepctr_torch/models/multitask/ple.py 95.87% <95.87%> (ø)
deepctr_torch/models/__init__.py 100.00% <100.00%> (ø)
deepctr_torch/models/dcnmix.py 88.37% <100.00%> (ø)
deepctr_torch/models/dien.py 97.23% <100.00%> (ø)
deepctr_torch/models/multitask/__init__.py 100.00% <100.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.