Cleand up the data collection

data/.gitignore

@@ -2,3 +2,5 @@ tmp_*.pickle
 collected.pickle
 test.pickle
 collectedWithImage.pickle
+collected_and_filtered.pickle
+collected_routes_np.pickle

docker-compose.yaml

@@ -8,6 +8,7 @@ services:
     container_name: pyrate
     ports:
     - "8888:8888"
+    - "6006:6006"
     volumes:
     - ./experiments:/pyrate/experiments
     - ./data:/pyrate/data

pyrate/pyrate.dockerfile

@@ -9,7 +9,7 @@ RUN apt install graphviz -yqq
 
 # Install additinal requirements
 RUN pip3 install jupyter tensorflow tensorboard keras-tuner black[jupyter] mapply humanize jupyter-resource-usage
-RUN pip3 install tensorflow-addons
+RUN pip3 install tensorflow-addons pydot pydotplus
 RUN pip install jupyter_contrib_nbextensions
 RUN jupyter contrib nbextension install --sys-prefix
 RUN jupyter nbextension enable scratchpad/main --sys-prefix