tree: a3df5f767b6a0f554f4b6702283972bc213fae9f [path history] [tgz]
  1. get_data.sh
  2. README.md
  3. train.pl
perl-package/AI-MXNet/examples/sparse/matrix_factorization/README.md

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.

  • get_data.sh
  • perl train.pl
  • To compare the training speed with (dense) Embedding, run perl train.pl --use-dense 1
  • To run the example on the GPU, run perl train.pl --use-gpu 1