Skip to content

YW81/wasserstein-dictionary-learning

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Wasserstein Dictionary Learning and Non-negative Matrix Factorization

A matlab toolbox to perform Wasserstein Dictionary Learning or NMF

This repository contains a matlab implementation of the methods described in the article:

Antoine Rolet, Marco Cuturi and Gabriel Peyré. Fast Dictionary Learning with a Smoothed Wasserstein Loss. To appear in Proceedings of the 19th International Conference on Artificial Intelligence and Statistics (AISTATS) 2016, Cadiz, Spain. JMLR: W&CP volume 41.

It also includes a script to reproduce the experiment in the introduction of the article step by step.

See LICENCE.txt for copyright and warranty information.

Getting started

The easiest way to start using Wasserstein Dictionary Learning is to open up matlab and run all sections of the script test_wassersteinDL_mixture1d.m

The main function is wasserstein_DL. Type doc wasserstein_DL in a matlab terminal for usage information

About

A matlab toolbox to perform Wasserstein Dictionary Learning or NMF

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MATLAB 76.2%
  • HTML 23.6%
  • M 0.2%