aboutsummaryrefslogtreecommitdiff
path: root/data.py
diff options
context:
space:
mode:
Diffstat (limited to 'data.py')
-rw-r--r--data.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/data.py b/data.py
index d2c4f77..7708863 100644
--- a/data.py
+++ b/data.py
@@ -1,6 +1,7 @@
import ast, csv
import socket
import fuel
+import numpy
from enum import Enum
from fuel.datasets import Dataset
from fuel.streams import DataStream
@@ -13,6 +14,9 @@ else:
client_ids = {int(x): y+1 for y, x in enumerate(open(DATA_PATH+"/client_ids.txt"))}
+porto_center = numpy.array([[ -8.61612, 41.1573]], dtype='float32')
+data_std = numpy.sqrt(numpy.array([[ 0.00333233, 0.00549598]], dtype='float32'))
+
class CallType(Enum):
CENTRAL = 0
STAND = 1