Update gen_sample

Author: yuta-nakahara

bayesml/contexttree/_contexttree.py

@@ -181,10 +181,8 @@ def _gen_sample_recursion(self,node,x):
             1 dimensional array whose elements are 0 or 1.
         """
         if node.leaf:  # 葉ノード
-            print('')
             return self.rng.choice(self.c_k,p=node.theta_vec)
         else:
-            print(f'{x[-node.depth-1]}',end=',')
             return self._gen_sample_recursion(node.children[x[-node.depth-1]],x)
 
     def _visualize_model_recursion(self,tree_graph,node,node_id,parent_id,sibling_num,p_v):
@@ -350,7 +348,7 @@ def gen_sample(self,sample_length,initial_values=None):
             x[:self.c_d_max] = initial_values
 
         for i in range(self.c_d_max,sample_length+self.c_d_max):
-            x[i] = self._gen_sample_recursion(self.root,x[:i])
+            x[i] = self._gen_sample_recursion(self.root,x[i-self.c_d_max:i])
 
         return x[self.c_d_max:]
 

bayesml/contexttree/_contexttree_test.py

@@ -1,7 +1,7 @@
 import numpy as np
 import bayesml.contexttree as model
 
-gen_model = model.GenModel(c_k=2,c_d_max=3,h_g=0.75)
+gen_model = model.GenModel(c_k=2,c_d_max=3,h_g=0.99)
 # print(gen_model.get_params())
 gen_model.gen_params()
 gen_model.visualize_model()