blob: 3ada5e80150a5db3c0c9528ac1892a109f862200 [file] [log] [blame] [view]
Matrix Factorization w/ Sparse Embedding
===========
The example demonstrates the basic usage of the SparseEmbedding operator in MXNet, adapted based on @leopd's recommender examples.
The operator is available on both CPU and GPU. This is for demonstration purpose only.
- `python train.py`
- To compare the training speed with (dense) Embedding, run `python train.py --use-dense`
- To run the example on the GPU, run `python train.py --use-gpu`