/kdcode-lm

Code for Learning K-way D-dimensional Discrete Codes For Compact Embedding Representations

Primary LanguagePython

This repository contains code for paper:

Ting Chen, Martin Renqiang Min, Yizhou Sun, "Learning K-way D-dimensional Discrete Codes For Compact Embedding Representations" Thirty-fifth International Conference on Machine Learning (ICML'18), Stockholm, Sweden, July. 2018.

To run the code, enter scripts folder, execute *.sh, results (log, as well as checkpoints etc.) will be saved into the results folder.

The code is based on tensorflow/models. Tensorflow 1.4 was originally used, but it seems the code also works in higher versions.