emd 0.1.1

A library for computing the Earth Mover's Distance
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

import numpy as np

from emd import emd

X = np.array([[0.0, 10.0],
              [1.0, 0.0]])

Y = np.array([[ 0.0, -1.0],
              [-1.0,  0.0]])

Z = np.array([[ 0.0, -1.0],
              [-1.0, 0.0],
              [0.5, 5.0]])

print emd(X, Y)

dist, flows = emd(X, Y, return_flows=True)
print flows

dist, flows = emd(X, Z, return_flows=True)
print flows