tensorlogic_train/hyperparameter/acquisition.rs
1//! Acquisition functions for Bayesian Optimization.
2
3/// Acquisition function type for Bayesian Optimization.
4#[derive(Debug, Clone, Copy, PartialEq)]
5pub enum AcquisitionFunction {
6 /// Expected Improvement - balances exploration and exploitation.
7 ExpectedImprovement { xi: f64 },
8 /// Upper Confidence Bound - uses uncertainty to guide exploration.
9 UpperConfidenceBound { kappa: f64 },
10 /// Probability of Improvement - probability of improving over best.
11 ProbabilityOfImprovement { xi: f64 },
12}
13
14impl Default for AcquisitionFunction {
15 fn default() -> Self {
16 Self::ExpectedImprovement { xi: 0.01 }
17 }
18}