From 12fbc9b96ea1ef2727c87c02fe8d2305235b4d15 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Wed, 29 Apr 2015 19:42:47 -0400 Subject: Change origin_call ids --- convert_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/convert_data.py b/convert_data.py index f81b40f..9684fa9 100755 --- a/convert_data.py +++ b/convert_data.py @@ -104,8 +104,9 @@ def convert(input_directory, save_path): h5file = h5py.File(save_path, 'w') split = {} split.update(read_stands(input_directory, h5file)) - split.update(read_taxis(input_directory, h5file, 'test', 'test_')) split.update(read_taxis(input_directory, h5file, 'train', '')) + print 'First origin_call not present in training set: ', len(origin_call_dict) + split.update(read_taxis(input_directory, h5file, 'test', 'test_')) split.update(unique(h5file)) h5file.attrs['split'] = H5PYDataset.create_split_array(split) h5file.flush() -- cgit v1.2.3 From a042000073bf348fddff23f41d30f49f2e874adb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Sat, 2 May 2015 05:17:11 +0000 Subject: Add equirectangular distance --- hdist.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/hdist.py b/hdist.py index 3f62eb3..4fc707e 100644 --- a/hdist.py +++ b/hdist.py @@ -2,6 +2,9 @@ from theano import tensor import theano import numpy +rearth = const(6371) +deg2rad = const(3.141592653589793 / 180) + def const(v): if theano.config.floatX == 'float32': return numpy.float32(v) @@ -9,9 +12,6 @@ def const(v): return numpy.float64(v) def hdist(a, b): - rearth = const(6371) - deg2rad = const(3.141592653589793 / 180) - lat1 = a[:, 0] * deg2rad lon1 = a[:, 1] * deg2rad lat2 = b[:, 0] * deg2rad @@ -27,5 +27,11 @@ def hdist(a, b): 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 -- cgit v1.2.3