Update ShareBottom

reczoo · Nov 5, 2024 · 85d993c · 85d993c
1 parent ebb37a1
commit 85d993c
Show file tree

Hide file tree

Showing 10 changed files with 13 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -88,7 +88,8 @@ Click-through rate (CTR) prediction is a critical task for various industrial ap
 | 50  | NeurIPS'22          | [APG](./model_zoo/APG)               | [APG: Adaptive Parameter Generation Network for Click-Through Rate Prediction](https://arxiv.org/abs/2203.16218) :triangular_flag_on_post:**Alibaba**                                |    [:arrow_upper_right:](https://github.com/reczoo/BARS/tree/main/ranking/ctr/APG)                                                                                                   | `torch`       |
 | 51  | KDD'23        | [PPNet](./model_zoo/PEPNet)             | [PEPNet: Parameter and Embedding Personalized Network for Infusing with Personalized Prior Information](https://arxiv.org/abs/2302.01115) :triangular_flag_on_post:**KuaiShou**                          |    [:arrow_upper_right:](https://github.com/reczoo/BARS/tree/main/ranking/ctr/PPNet)                                                                                                   | `torch`       |
 |<tr><th colspan=6 align="center">:open_file_folder: **Multi-Task Modeling**</th></tr>|
-| 52  |     MachineLearn'97      | [SharedBottom](./model_zoo/multitask/SharedBottom)               | [Multitask Learning](https://link.springer.com/article/10.1023/A:1007379606734)                                                                                            |                                                                                                                 | `torch`       |
+| 52  |     Arxiv'17      | [ShareBottom](./model_zoo/multitask/ShareBottom)               | [An Overview of Multi-Task Learning
+in Deep Neural Networks](https://arxiv.org/abs/1706.05098)                                                                                            |                                                                                                                 | `torch`       |
 | 53  | KDD'18          | [MMoE](./model_zoo/multitask/MMOE)               | [Modeling Task Relationships in Multi-task Learning with Multi-Gate Mixture-of-Experts](https://dl.acm.org/doi/pdf/10.1145/3219819.3220007) :triangular_flag_on_post:**Google**                                                                                            |                                                                                                                 | `torch`       |
 | 54  | KDD'18          | [PLE](./model_zoo/multitask/PLE)               | [Progressive Layered Extraction (PLE): A Novel Multi-Task Learning (MTL) Model for Personalized Recommendations](https://dl.acm.org/doi/10.1145/3383313.3412236) :triangular_flag_on_post:**Tencent**                                                                                            |                                                                                                                 | `torch`       |
 |<tr><th colspan=6 align="center">:open_file_folder: **Multi-Domain Modeling**</th></tr>|

diff --git a/model_zoo/multitask/SharedBottom/README.md → model_zoo/multitask/ShareBottom/README.md b/model_zoo/multitask/SharedBottom/README.md → model_zoo/multitask/ShareBottom/README.md
diff --git a/...k/SharedBottom/config/dataset_config.yaml → ...sk/ShareBottom/config/dataset_config.yaml b/...k/SharedBottom/config/dataset_config.yaml → ...sk/ShareBottom/config/dataset_config.yaml
diff --git a/...ask/SharedBottom/config/model_config.yaml → ...task/ShareBottom/config/model_config.yaml b/...ask/SharedBottom/config/model_config.yaml → ...task/ShareBottom/config/model_config.yaml
@@ -12,16 +12,16 @@ Base:
     feature_specs: null
     feature_config: null
 
-SharedBottom_test:
+ShareBottom_test:
     model: SharedBottom
     dataset_id: tiny_mtl
-    loss: ['binary_crossentropy','binary_crossentropy']
+    loss: ['binary_crossentropy', 'binary_crossentropy']
     metrics: ['logloss', 'AUC']
-    task: ['binary_classification','binary_classification']
+    task: ['binary_classification', 'binary_classification']
     num_tasks: 2
     optimizer: adam
     learning_rate: 1.e-3
-    bottom_hidden_units: [512,256,128]
+    bottom_hidden_units: [512, 256, 128]
     tower_hidden_units: [128, 64]
     hidden_activations: relu
     net_regularizer: 0
@@ -30,8 +30,8 @@ SharedBottom_test:
     net_dropout: 0
     batch_size: 128
     embedding_dim: 128
-    epochs: 50
+    epochs: 1
     shuffle: True
     seed: 2023
     monitor: 'AUC'
-    monitor_mode: 'max'
+    monitor_mode: 'max'
diff --git a/...multitask/SharedBottom/fuxictr_version.py → .../multitask/ShareBottom/fuxictr_version.py b/...multitask/SharedBottom/fuxictr_version.py → .../multitask/ShareBottom/fuxictr_version.py
diff --git a/...l_zoo/multitask/SharedBottom/run_expid.py → model_zoo/multitask/ShareBottom/run_expid.py b/...l_zoo/multitask/SharedBottom/run_expid.py → model_zoo/multitask/ShareBottom/run_expid.py
diff --git a/...ultitask/SharedBottom/src/SharedBottom.py → .../multitask/ShareBottom/src/ShareBottom.py b/...ultitask/SharedBottom/src/SharedBottom.py → .../multitask/ShareBottom/src/ShareBottom.py
@@ -22,7 +22,7 @@
 from fuxictr.pytorch.layers import FeatureEmbedding, MLP_Block
 
 
-class SharedBottom(MultiTaskModel):
+class ShareBottom(MultiTaskModel):
     def __init__(self,
                  feature_map,
                  model_id="SharedBottom",
@@ -40,7 +40,7 @@ def __init__(self,
                  embedding_regularizer=None,
                  net_regularizer=None,
                  **kwargs):
-        super(SharedBottom, self).__init__(feature_map,
+        super(ShareBottom, self).__init__(feature_map,
                                            task=task,
                                            loss_weight=loss_weight,
                                            num_tasks=num_tasks,

diff --git a/model_zoo/multitask/ShareBottom/src/__init__.py b/model_zoo/multitask/ShareBottom/src/__init__.py
@@ -0,0 +1 @@
+from .ShareBottom import *
diff --git a/model_zoo/multitask/SharedBottom/src/__init__.py b/model_zoo/multitask/SharedBottom/src/__init__.py
diff --git a/model_zoo/multitask/__init__.py b/model_zoo/multitask/__init__.py
@@ -1,2 +1,3 @@
 from .MMoE.src import MMoE
-from .SharedBottom.src import SharedBottom
+from .ShareBottom.src import ShareBottom
+from .PLE.src import PLE