From 1984daacfdb69b1f2a9f5919ae7c8609f761153f Mon Sep 17 00:00:00 2001 From: Yili Li Date: Mon, 16 Oct 2023 18:38:34 +0800 Subject: [PATCH 1/2] add xdeepfm I85WPL add xdeepfm train and infer scripts Signed-off-by: Yili Li --- .../ctr/xdeepfm/paddlepaddle/README.md | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 recommendation/ctr/xdeepfm/paddlepaddle/README.md diff --git a/recommendation/ctr/xdeepfm/paddlepaddle/README.md b/recommendation/ctr/xdeepfm/paddlepaddle/README.md new file mode 100644 index 000000000..55ae5d629 --- /dev/null +++ b/recommendation/ctr/xdeepfm/paddlepaddle/README.md @@ -0,0 +1,27 @@ +# xDeepFM + +## Step 1: Installing + +``` +git clone --recursive https://github.com/PaddlePaddle/PaddleRec.git +cd PaddleRec +pip3 install -r requirements.txt +``` + +## Step 2: Download data + +``` +cd PaddleRec +cd datasets/criteo/ +sh run.sh +``` + +## Step 3: Run xDeepFM + +``` +cd models/rank/xdeepfm +# train +python3 -u ../../../tools/trainer.py -m config_bigdata.yaml +# Eval +python3 -u ../../../tools/infer.py -m config_bigdata.yaml +``` -- Gitee From af68062ab6804551e50dc15ac60db324cd0a139a Mon Sep 17 00:00:00 2001 From: Yili Li Date: Mon, 23 Oct 2023 13:44:45 +0800 Subject: [PATCH 2/2] fix readme --- recommendation/ctr/xdeepfm/paddlepaddle/README.md | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/recommendation/ctr/xdeepfm/paddlepaddle/README.md b/recommendation/ctr/xdeepfm/paddlepaddle/README.md index 55ae5d629..32dc4bb7e 100644 --- a/recommendation/ctr/xdeepfm/paddlepaddle/README.md +++ b/recommendation/ctr/xdeepfm/paddlepaddle/README.md @@ -1,5 +1,6 @@ # xDeepFM - +## Model description +xDeepFM propose a novel network named Compressed Interaction Network (CIN), which aims to learn high-order feature interactions explicitly. xDeepFM can automatically learn high-order feature interactions in both explicit and implicit fashions, which is of great significance to reducing manual feature engineering work. ## Step 1: Installing ``` @@ -8,7 +9,7 @@ cd PaddleRec pip3 install -r requirements.txt ``` -## Step 2: Download data +## Step 2: Preparing datasets ``` cd PaddleRec @@ -16,7 +17,7 @@ cd datasets/criteo/ sh run.sh ``` -## Step 3: Run xDeepFM +## Step 3: Training ``` cd models/rank/xdeepfm @@ -25,3 +26,8 @@ python3 -u ../../../tools/trainer.py -m config_bigdata.yaml # Eval python3 -u ../../../tools/infer.py -m config_bigdata.yaml ``` + +## Result +| GPUs | IPS | AUC | +|-------------|-----------|-------------| +| BI-V100x1 | 6000 | 0.7955 | -- Gitee