Merge pull request #43 from dityas/dev

Dev
dityas · Jan 10, 2020 · 7a932a1 · 7a932a1
2 parents 8b394b0 + 19970aa
commit 7a932a1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 9 deletions.
diff --git a/Protos/build/Protos.jar b/Protos/build/Protos.jar
diff --git a/Protos/src/thinclab/representations/belieftreerepresentations/DynamicBeliefTree.java b/Protos/src/thinclab/representations/belieftreerepresentations/DynamicBeliefTree.java
@@ -82,17 +82,27 @@ public void buildTree() {
 		List<Integer> prevNodes = new ArrayList<Integer>();
 		prevNodes.addAll(leafNodes);
 
-		for (int t = 1; t < this.maxT; t++) {
-
-			List<Integer> nextNodes = this.getNextPolicyNodes(prevNodes, t);
-			prevNodes = nextNodes;
-
-			if (t == 1) {
-				this.leafNodes.clear();
-				this.leafNodes.addAll(prevNodes);
-			}
+		/*
+		 * if one step look ahead, don't add the next leaves to models
+		 * but store them for computing the next step
+		 */
+		if (this.maxT == 1) {
+			this.leafNodes.clear();
+			this.leafNodes.addAll(this.getNextPolicyNodes(prevNodes, 1));
 		}
 
+		else {
+			for (int t = 1; t < this.maxT; t++) {
+
+				List<Integer> nextNodes = this.getNextPolicyNodes(prevNodes, t);
+				prevNodes = nextNodes;
+
+				if (t == 1) {
+					this.leafNodes.clear();
+					this.leafNodes.addAll(prevNodes);
+				}
+			}
+		}
 	}
 
 	// ----------------------------------------------------------------------------------------