It is Gaussian process regression implementations with TensorFlow. numpy.ndarray, Y_train: numpy.ndarray, prior_mu: Optional[callable], str_cov: str, fix_noise: bool = True, num_iters: int = 1000, debug: bool = False) → Tuple[numpy.ndarray, numpy.ndarray, dict]

This function computes the kernel matrix optimized by optimization method specified, its inverse matrix, and the optimized hyperparameters, using TensorFlow and TensorFlow probability.

  • X_train (numpy.ndarray) – inputs. Shape: (n, d) or (n, m, d).
  • Y_train (numpy.ndarray) – outputs. Shape: (n, 1).
  • prior_mu (function or NoneType) – prior mean function or None.
  • str_cov (str.) – the name of covariance function.
  • fix_noise (bool., optional) – flag for fixing a noise.
  • num_iters (int., optional) – the number of iterations for optimizing negative log likelihood.
  • debug (bool., optional) – flag for printing log messages.

a tuple of kernel matrix over X_train, kernel matrix inverse, and dictionary of hyperparameters.

Return type:

tuple of (numpy.ndarray, numpy.ndarray, dict.)


AssertionError, ValueError