jeremymiller00 / PyWMD

Word Mover's Distance (WMD) in Python

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Word Mover's Distance (WMD) for Python

This is the implementation of Word Mover's Distance (WMD) in the following paper:

Matt J. Kusner, Yu Sun, Nicholas I. Kolkin, Kilian Q. Weinberger, "From Word Embeddings to Document Distances," ICML (2015).

WMD is a special case of Wasserstein distance.

Related blog post: Word Mover’s Distance as a Linear Programming Problem

About

Word Mover's Distance (WMD) in Python


Languages

Language:Jupyter Notebook 87.4%Language:Python 12.6%