Add normalization

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