This section shows you how to get your feet wet with BoTorch.
Before jumping the gun, we recommend you start with the high-level Overview to learn about the basic concepts in BoTorch.
BoTorch is easily installed via
Anaconda (strongly recommended for OSX)
conda install botorch -c pytorch -c gpytorch -c conda-forge
pip install botorch
For more installation options and detailed instructions, please see the Project Readme on GitHub.
Here's a quick run down of the main components of a Bayesian Optimization loop.
Fit a Gaussian Process model to data
import torch from botorch.models import SingleTaskGP from botorch.fit import fit_gpytorch_mll from gpytorch.mlls import ExactMarginalLogLikelihood from botorch.models.transforms.outcome import Standardize train_X = torch.rand(10, 2, dtype=torch.float64) # explicit output dimension -- Y is 10 x 1 train_Y = 1 - (train_X - 0.5).norm(dim=-1, keepdim=True) train_Y += 0.1 * torch.rand_like(train_Y) gp = SingleTaskGP(train_X, trainb_Y, outcome_transform=Standardize(m=1)) mll = ExactMarginalLogLikelihood(gp.likelihood, gp) fit_gpytorch_mll(mll)
Construct an acquisition function
from botorch.acquisition import UpperConfidenceBound UCB = UpperConfidenceBound(gp, beta=0.1)
Optimize the acquisition function
from botorch.optim import optimize_acqf bounds = torch.stack([torch.zeros(2), torch.ones(2)]) candidate, acq_value = optimize_acqf( UCB, bounds=bounds, q=1, num_restarts=5, raw_samples=20, )
Our Jupyter notebook tutorials help you get off the ground with BoTorch. View and download them here.
For an in-depth reference of the various BoTorch internals, see our API Reference.
You'd like to contribute to BoTorch? Great! Please see here for how to help out.