Merge pull request #41 from yuta-nakahara/develop-contexttree-GenModel

Develop contexttree gen model

Author: yuta-nakahara

bayesml/_check.py

@@ -1,6 +1,8 @@
 # Code Author
 # Yuta Nakahara <yuta.nakahara@aoni.waseda.jp>
 # Yuji Iikubo <yuji-iikubo.8@fuji.waseda.jp>
+# Yasushi Esaki <esakiful@gmail.com>
+# Jun Nishikawa <jun.b.nishikawa@gmail.com>
 import numpy as np
 
 _EPSILON = np.sqrt(np.finfo(np.float64).eps)
@@ -179,6 +181,14 @@ def float_vec_sum_1(val,val_name,exception_class):
             return val
     raise(exception_class(val_name + " must be a 1-dimensional numpy.ndarray, and the sum of its elements must equal to 1."))
 
+def float_vecs_sum_1(val,val_name,exception_class):
+    if type(val) is np.ndarray:
+        if np.issubdtype(val.dtype,np.integer) and val.ndim >= 1 and np.all(np.abs(np.sum(val, axis=-1) - 1.) <= _EPSILON):
+            return val.astype(float)
+        if np.issubdtype(val.dtype,np.floating) and val.ndim >= 1 and np.all(np.abs(np.sum(val, axis=-1) - 1.) <= _EPSILON):
+            return val
+    raise(exception_class(val_name + " must be a numpy.ndarray whose ndim >= 1, and the sum along the last dimension must equal to 1."))
+
 def int_(val,val_name,exception_class):   
     if np.issubdtype(type(val),np.integer):
         return val
@@ -205,3 +215,9 @@ def onehot_vecs(val,val_name,exception_class):
         if np.issubdtype(val.dtype,np.integer) and val.ndim >= 1 and np.all(val >= 0) and np.all(val.sum(axis=-1)==1):
             return val
     raise(exception_class(val_name + " must be a numpy.ndarray whose dtype is int and whose last axis constitutes one-hot vectors."))
+
+def shape_consistency(val: int, val_name: str, correct: int, correct_name: str, exception_class):
+    if val != correct:
+        message = (f"{val_name} must coincide with {correct_name}: "
+                   + f"{val_name} = {val}, {correct_name} = {correct}")
+        raise(exception_class(message))

bayesml/base.py

@@ -257,13 +257,21 @@ def load_hn_params(self,filename):
             +'or ``LearnModel.save_hn_params()``.')
             )
 
-    @abstractmethod
     def reset_hn_params(self):
-        pass
-
-    @abstractmethod
+        """Reset the hyperparameters of the posterior distribution to their initial values.
+        
+        They are reset to the output of `self.get_h0_params()`.
+        Note that the parameters of the predictive distribution are also calculated from them.
+        """
+        self.set_hn_params(*self.get_h0_params().values())
+    
     def overwrite_h0_params(self):
-        pass
+        """Overwrite the initial values of the hyperparameters of the posterior distribution by the learned values.
+        
+        They are overwitten by the output of `self.get_hn_params()`.
+        Note that the parameters of the predictive distribution are also calculated from them.
+        """
+        self.set_h0_params(*self.get_hn_params().values())
 
     @abstractmethod
     def update_posterior(self):

bayesml/contexttree/__init__.py

@@ -0,0 +1,4 @@
+from ._contexttree import GenModel
+# from ._contexttree import LearnModel
+
+__all__ = ["GenModel"]#, "LearnModel"]