add constant mean function for gpr

pymc-learn · Emaasit · Dec 3, 2018 · Dec 3, 2018 · Dec 3, 2018 · abde49e7e84fbe52849fe6ccebcedc9d85564d1c
commit abde49e7e84fbe52849fe6ccebcedc9d85564d1c
diff --git a/pmlearn/__init__.py b/pmlearn/__init__.py
@@ -10,7 +10,7 @@
 See http://pymc-learn.org for complete documentation.
 """

-__version__ = '0.0.1.rc2'
+__version__ = '0.0.1.rc3'

 __all__ = ['gaussian_process',
           'linear_model',

diff --git a/pmlearn/gaussian_process/gpr.py b/pmlearn/gaussian_process/gpr.py
@@ -144,7 +144,7 @@ def create_model(self):
            if self.prior_mean is None:
                mean_function = pm.gp.mean.Zero()
            else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)

            self.gp = pm.gp.Latent(mean_func=mean_function,
                                   cov_func=cov_function)
@@ -259,7 +259,7 @@ def create_model(self):
            if self.prior_mean is None:
                mean_function = pm.gp.mean.Zero()
            else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)

            self.gp = pm.gp.Latent(mean_func=mean_function,
                                   cov_func=cov_function)
@@ -373,7 +373,7 @@ def create_model(self):
            if self.prior_mean is None:
                mean_function = pm.gp.mean.Zero()
            else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)

            self.gp = pm.gp.MarginalSparse(mean_func=mean_function,
                                           cov_func=cov_function,