提交 2a2ad7b5 编写于 作者: F frankwhzhang

mmoe readme

上级 a5015123
mkdir train_data
mkdir test_data
mkdir data
train_path="data/census-income.data"
test_path="data/census-income.test"
train_data_path="train_data/"
test_data_path="test_data/"
pip install -r requirements.txt
wget -P data/ https://archive.ics.uci.edu/ml/machine-learning-databases/census-income-mld/census.tar.gz
tar -zxvf data/census.tar.gz -C data/
python data_preparation.py --train_path ${train_path} \
--test_path ${test_path} \
--train_data_path ${train_data_path}\
--test_data_path ${test_data_path}
......@@ -14,6 +14,19 @@ python -m paddlerec.run -m paddlerec.models.multitask.mmoe
根据原论文,我们在开源数据集Census-income Data上验证模型效果
### 数据下载及预处理
数据地址: [Census-income Data](https://archive.ics.uci.edu/ml/machine-learning-databases/census-income-mld/census.tar.gz )
数据解压后, 在data/run.sh脚本文件中添加文件的路径,并运行脚本。
```shell
cd data
sh run.sh
```
脚本运行后,在config.yaml中修改数据路径dataset.data_path
### 参数
config.yaml中的hyper_parameters部分,batch_size:32, epochs:400
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册