apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-27 14:25:16 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

doctest on TanhFunction

Browse source
This commit is contained in:
beckdaniel 2016-02-24 11:32:32 +00:00
parent b266b578cd
commit b3f8f263da
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
GPy/util/warping_functions.py
View file

@ -50,7 +50,12 @@ class WarpingFunction(Parameterized):
class TanhFunction(WarpingFunction): class TanhFunction(WarpingFunction):
"""
This is the function proposed in Snelson et al.:
A sum of tanh functions with linear trends outside
the range. Notice the term 'd', which scales the
linear trend.
"""
def __init__(self, n_terms=3, initial_y=None): def __init__(self, n_terms=3, initial_y=None):
""" """
n_terms specifies the number of tanh terms to be used n_terms specifies the number of tanh terms to be used
@ -58,11 +63,9 @@ class TanhFunction(WarpingFunction):
self.n_terms = n_terms self.n_terms = n_terms
self.num_parameters = 3 * self.n_terms + 1 self.num_parameters = 3 * self.n_terms + 1
self.psi = np.ones((self.n_terms, 3)) self.psi = np.ones((self.n_terms, 3))
super(TanhFunction, self).__init__(name='warp_tanh') super(TanhFunction, self).__init__(name='warp_tanh')
self.psi = Param('psi', self.psi) self.psi = Param('psi', self.psi)
self.psi[:, :2].constrain_positive() self.psi[:, :2].constrain_positive()
self.d = Param('%s' % ('d'), 1.0, Logexp()) self.d = Param('%s' % ('d'), 1.0, Logexp())
self.link_parameter(self.psi) self.link_parameter(self.psi)
self.link_parameter(self.d) self.link_parameter(self.d)