Browse Source

update readme

tags/v0.6.0-beta
hanjun996 5 years ago
parent
commit
f64352534b
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      model_zoo/wide_and_deep/README.md

+ 11
- 0
model_zoo/wide_and_deep/README.md View File

@@ -4,6 +4,17 @@ This is an implementation of WideDeep as described in the [Wide & Deep Learning


WideDeep model jointly trained wide linear models and deep neural network, which combined the benefits of memorization and generalization for recommender systems. WideDeep model jointly trained wide linear models and deep neural network, which combined the benefits of memorization and generalization for recommender systems.


## Requirements

- Install [MindSpore](https://www.mindspore.cn/install/en).

- Download the dataset and convert the dataset to mindrecord, command as follows:
```
python src/preprocess_data.py
```
Arguments:
* `--data_path`: Dataset storage path (Default: ./criteo_data/).
## Dataset ## Dataset
The Criteo datasets are used for model training and evaluation. The Criteo datasets are used for model training and evaluation.




Loading…
Cancel
Save