Any-to-Any
fuxictr

Add pipeline tag and library name

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +55 -3
README.md CHANGED
@@ -1,3 +1,55 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: any-to-any
4
+ library_name: fuxictr
5
+ ---
6
+
7
+ ## Official implementation code of DISCO.AHU team
8
+ ## 🔥 Winning 2nd Place in WWW2025 Multimodal CTR Prediction Challenge Track
9
+
10
+ ### 🔥 Follow to perfectly reproduce the results of this code.
11
+
12
+ - To facilitate reproducibility, we share the model checkpoints on Hugging Face: https://huggingface.co/salmon1802/QIN/tree/main
13
+ - In ./checkpoints and ./submission folders have our run logs and submission files, respectively.
14
+ - This submission can be reproduced manually by following the actions below, or by directly using the one-click run script run.sh
15
+
16
+ ### Data Preparation
17
+
18
+ 1. Download the datasets at: https://recsys.westlake.edu.cn/MicroLens_1M_MMCTR
19
+
20
+ 2. Unzip the data files to the `data` directory
21
+
22
+ ```bash
23
+ cd ./data/
24
+ wget -r -np -nH --cut-dirs=1 http://recsys.westlake.edu.cn/MicroLens_1M_MMCTR/MicroLens_1M_x1/
25
+ ```
26
+
27
+ ### Environment
28
+
29
+ We run the experiments on RTX 4090 GPU of AutoDL.com
30
+
31
+ Please set up the environment as follows.
32
+
33
+ + torch==2.0.0+cu118
34
+ + fuxictr==2.3.7
35
+
36
+ ```
37
+ conda create -n fuxictr_www python==3.8
38
+ pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
39
+ pip install -r requirements.txt
40
+ source activate fuxictr_www
41
+ ```
42
+
43
+ ### How to Run
44
+
45
+ Train the model on train and validation sets:
46
+
47
+ ```
48
+ python run_expid.py
49
+ ```
50
+ The parameters QIN_variety_v9 in __./config/qin_config/model_config.yaml__ are set to the optimal hyperparameters in the environment described above.
51
+
52
+ #### Tips
53
+ It is worth mentioning that after our tests, we find that although the parameter num_row = 4 achieves the best performance in the above environments, there is training instability in some environments.
54
+
55
+ When this happens, we suggest that sacrificing some performance in favor of setting num_row = 3 reproduces the results well.