aboutsummaryrefslogtreecommitdiff
path: root/hdist.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-05 14:15:21 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-05 14:15:21 -0400
commit54613c1f9cf510ca7a71d6619418f2247515aec6 (patch)
treebed9a5a11ef5b7feecee44095a29400e32f76b05 /hdist.py
parent712035b88be1816d3fbd58ce69ae6464767c780e (diff)
downloadtaxi-54613c1f9cf510ca7a71d6619418f2247515aec6.tar.gz
taxi-54613c1f9cf510ca7a71d6619418f2247515aec6.zip
Add models for time predictioAdd models for time prediction
Diffstat (limited to 'hdist.py')
-rw-r--r--hdist.py37
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