diff options
author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 14:15:21 -0400 |
---|---|---|
committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 14:15:21 -0400 |
commit | 54613c1f9cf510ca7a71d6619418f2247515aec6 (patch) | |
tree | bed9a5a11ef5b7feecee44095a29400e32f76b05 /hdist.py | |
parent | 712035b88be1816d3fbd58ce69ae6464767c780e (diff) | |
download | taxi-54613c1f9cf510ca7a71d6619418f2247515aec6.tar.gz taxi-54613c1f9cf510ca7a71d6619418f2247515aec6.zip |
Add models for time predictioAdd models for time prediction
Diffstat (limited to 'hdist.py')
-rw-r--r-- | hdist.py | 37 |
1 files changed, 0 insertions, 37 deletions
diff --git a/hdist.py b/hdist.py deleted file mode 100644 index 866aea4..0000000 --- a/hdist.py +++ /dev/null @@ -1,37 +0,0 @@ -from theano import tensor -import theano -import numpy - -def const(v): - if theano.config.floatX == 'float32': - return numpy.float32(v) - else: - return numpy.float64(v) - -rearth = const(6371) -deg2rad = const(3.141592653589793 / 180) - -def hdist(a, b): - lat1 = a[:, 0] * deg2rad - lon1 = a[:, 1] * deg2rad - lat2 = b[:, 0] * deg2rad - lon2 = b[:, 1] * deg2rad - - dlat = abs(lat1-lat2) - dlon = abs(lon1-lon2) - - al = tensor.sin(dlat/2)**2 + tensor.cos(lat1) * tensor.cos(lat2) * (tensor.sin(dlon/2)**2) - d = tensor.arctan2(tensor.sqrt(al), tensor.sqrt(const(1)-al)) - - hd = const(2) * rearth * d - - return tensor.switch(tensor.eq(hd, float('nan')), (a-b).norm(2, axis=1), hd) - -def erdist(a, b): - lat1 = a[:, 0] * deg2rad - lon1 = a[:, 1] * deg2rad - lat2 = b[:, 0] * deg2rad - lon2 = b[:, 1] * deg2rad - x = (lon2-lon1) * tensor.cos((lat1+lat2)/2) - y = (lat2-lat1) - return tensor.sqrt(tensor.sqr(x) + tensor.sqr(y)) * rearth |