eric-haibin-lin closed pull request #9303: fix wide&deep URL: https://github.com/apache/incubator-mxnet/pull/9303
This is a PR merged from a forked repository. As GitHub hides the original diff on merge, it is displayed below for the sake of provenance: As this is a foreign pull request (from a fork), the diff is supplied below (as it won't show otherwise due to GitHub magic): diff --git a/example/sparse/wide_deep/README.md b/example/sparse/wide_deep/README.md index a538106216..3df5e420ee 100644 --- a/example/sparse/wide_deep/README.md +++ b/example/sparse/wide_deep/README.md @@ -4,4 +4,4 @@ The example demonstrates how to train [wide and deep model](https://arxiv.org/ab The final accuracy should be around 85%. -- `python wide_deep_classification.py` +- `python train.py` diff --git a/example/sparse/wide_deep/model.py b/example/sparse/wide_deep/model.py index e8ba5318b5..b90745599c 100644 --- a/example/sparse/wide_deep/model.py +++ b/example/sparse/wide_deep/model.py @@ -49,9 +49,9 @@ def wide_deep_model(num_linear_features, num_embed_features, num_cont_features, hidden = mx.symbol.concat(*features, dim=1) hidden = mx.symbol.FullyConnected(data=hidden, num_hidden=hidden_units[1]) - hideen = mx.symbol.Activation(data=hidden, act_type='relu') + hidden = mx.symbol.Activation(data=hidden, act_type='relu') hidden = mx.symbol.FullyConnected(data=hidden, num_hidden=hidden_units[2]) - hideen = mx.symbol.Activation(data=hidden, act_type='relu') + hidden = mx.symbol.Activation(data=hidden, act_type='relu') deep_out = mx.symbol.FullyConnected(data=hidden, num_hidden=2) out = mx.symbol.SoftmaxOutput(linear_out + deep_out, label, name='model') ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
