#![warn(missing_docs)]
pub mod automatic_kernel;
pub mod bayesian_optimization;
pub mod fitc;
pub mod gpc;
pub mod gpr;
pub mod kernel_optimization;
pub mod kernel_selection;
pub mod kernel_structure_learning;
pub mod kernels;
pub mod marginal_likelihood;
pub mod multi_output;
pub mod nystrom;
pub mod random_fourier_features;
pub mod sparse_gpr;
pub mod sparse_spectrum;
pub mod structured_kernel_interpolation;
pub mod utils;
pub mod variational;
pub use automatic_kernel::{
AutomaticKernelConstructor, DataCharacteristics, KernelConstructionResult,
};
pub use bayesian_optimization::{AcquisitionFunction, BayesianOptimizer, BayesianOptimizerFitted};
pub use fitc::{
FitcGaussianProcessRegressor, FitcGaussianProcessRegressorConfig, FitcGprTrained,
InducingPointInit as FitcInducingPointInit,
};
pub use gpr::{GaussianProcessRegressor, GprTrained};
pub use gpc::{
ExpectationPropagationGaussianProcessClassifier, EpGpcTrained,
GaussianProcessClassifier, GpcConfig, GpcTrained,
MultiClassGaussianProcessClassifier, McGpcTrained,
SparseGaussianProcessClassifier, SgpcTrained,
VariationalGaussianProcessClassifier, VariationalGpcConfig, VariationalGpcTrained,
};
pub use kernel_optimization::OptimizationResult as KernelOptimizationResult;
pub use kernel_optimization::{optimize_kernel_parameters, KernelOptimizer};
pub use kernel_selection::{
select_best_kernel, select_kernel_aic, select_kernel_bic, select_kernel_cv,
KernelSelectionConfig, KernelSelectionResult, KernelSelector, SelectionCriterion,
};
pub use kernel_structure_learning::{
ConvergenceInfo, KernelGrammar, KernelStructureLearner, NonTerminalOperation, SearchStrategy,
StructureLearningResult, TerminalKernel,
};
pub use marginal_likelihood::OptimizationResult as MarginalLikelihoodOptimizationResult;
pub use marginal_likelihood::{
cross_validate_hyperparameters, log_marginal_likelihood, log_marginal_likelihood_stable,
optimize_hyperparameters, MarginalLikelihoodOptimizer,
};
pub use multi_output::{
LinearModelCoregionalization, LmcTrained,
MultiOutputGaussianProcessRegressor, MogprTrained,
};
pub use nystrom::{LandmarkSelection, NystromGaussianProcessRegressor, NystromGprTrained};
pub use random_fourier_features::{
RandomFourierFeatures, RffFitted,
RandomFourierFeaturesGPR, RffGprTrained,
};
pub use sparse_gpr::{InducingPointInit, SgprTrained, SparseGaussianProcessRegressor};
pub use sparse_spectrum::{
SparseSpectrumGaussianProcessRegressor, SparseSpectrumGprTrained, SpectralApproximationInfo,
SpectralSelectionMethod,
};
pub use structured_kernel_interpolation::{
GridBoundsMethod, InterpolationMethod, SkiApproximationInfo, SkiGprTrained,
StructuredKernelInterpolationGPR,
};
pub use variational::{
VariationalOptimizer, VariationalSparseGaussianProcessRegressor, VsgprTrained,
};
pub use kernels::{Kernel, RBF};
pub use utils::{robust_cholesky, triangular_solve, matrix_inverse};