From c47472838a24a6943e2a8e17fd1b13b2aa505a64 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Wed, 3 Jul 2019 13:04:35 -0400
Subject: [PATCH] moved split_tree into baseline repo
---
flatland/envs/observations.py | 36 -----------------------------------
notebooks/Scene_Editor.ipynb | 2 +-
2 files changed, 1 insertion(+), 37 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 8c3f260b..03b8d56e 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -557,42 +557,6 @@ class TreeObsForRailEnv(ObservationBuilder):
prompt=prompt_[children],
current_depth=current_depth + 1)
- def split_tree(self, tree, num_features_per_node=8, current_depth=0):
- """
-
- :param tree:
- :param num_features_per_node:
- :param prompt:
- :param current_depth:
- :return:
- """
-
- if len(tree) < num_features_per_node:
- return [], [], []
-
- depth = 0
- tmp = len(tree) / num_features_per_node - 1
- pow4 = 4
- while tmp > 0:
- tmp -= pow4
- depth += 1
- pow4 *= 4
- child_size = (len(tree) - num_features_per_node) // 4
- tree_data = tree[:4].tolist()
- distance_data = [tree[4]]
- agent_data = tree[5:num_features_per_node].tolist()
- for children in range(4):
- child_tree = tree[(num_features_per_node + children * child_size):
- (num_features_per_node + (children + 1) * child_size)]
- tmp_tree_data, tmp_distance_data, tmp_agent_data = self.split_tree(child_tree,
- num_features_per_node,
- current_depth=current_depth + 1)
- if len(tmp_tree_data) > 0:
- tree_data.extend(tmp_tree_data)
- distance_data.extend(tmp_distance_data)
- agent_data.extend(tmp_agent_data)
- return tree_data, distance_data, agent_data
-
def _set_env(self, env):
self.env = env
if self.predictor:
diff --git a/notebooks/Scene_Editor.ipynb b/notebooks/Scene_Editor.ipynb
index abfe6220..a36b4e15 100644
--- a/notebooks/Scene_Editor.ipynb
+++ b/notebooks/Scene_Editor.ipynb
@@ -111,7 +111,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
- "version": "3.6.8"
+ "version": "3.6.5"
},
"latex_envs": {
"LaTeX_envs_menu_present": true,
--
GitLab