models.py 文件源码

def __init__(self, kernel='rbf', nbases=50, lenscale=1., var=1.,
                 falloff=1., regulariser=1., ard=True,
                 indicator_field='censored', maxiter=3000,
                 batch_size=10, alpha=0.01, beta1=0.9,
                 beta2=0.99, epsilon=1e-8, random_state=None):

        lhood = Switching(lenscale=falloff,
                          var_init=Parameter(var, Positive()))

        super().__init__(likelihood=lhood,
                         basis=None,
                         maxiter=maxiter,
                         batch_size=batch_size,
                         updater=Adam(alpha, beta1, beta2, epsilon),
                         random_state=random_state
                         )

        self.indicator_field = indicator_field
        self._store_params(kernel, regulariser, nbases, lenscale, ard)