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.

## 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
The Criteo datasets are used for model training and evaluation.



Loading…
Cancel
Save