From f64352534b6b2d86dc726d1056a432247e04ca9d Mon Sep 17 00:00:00 2001 From: hanjun996 Date: Thu, 16 Jul 2020 16:32:31 +0800 Subject: [PATCH] update readme --- model_zoo/wide_and_deep/README.md | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/model_zoo/wide_and_deep/README.md b/model_zoo/wide_and_deep/README.md index 000e6a5335..b91c39ab30 100644 --- a/model_zoo/wide_and_deep/README.md +++ b/model_zoo/wide_and_deep/README.md @@ -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.