AcquisitionFunctionOption.h

Go to the documentation of this file.

 
#pragma once
 
#include <set>
 
#include "autopas/options/Option.h"
 
namespace autopas {
inline namespace options {
class AcquisitionFunctionOption : public Option<AcquisitionFunctionOption> {
 public:
  enum Value {
    upperConfidenceBound,
    mean,
    variance,
    probabilityOfImprovement,
    expectedImprovement,
  };
 
  AcquisitionFunctionOption() = default;
 
  constexpr AcquisitionFunctionOption(Value option) : _value(option) {}
 
  constexpr operator Value() const { return _value; }
 
  static std::set<AcquisitionFunctionOption> getDiscouragedOptions() { return {}; }
 
  static std::map<AcquisitionFunctionOption, std::string> getOptionNames() {
    return {
        {AcquisitionFunctionOption::upperConfidenceBound, "upper-confidence-bound"},
        {AcquisitionFunctionOption::mean, "mean"},
        {AcquisitionFunctionOption::variance, "variance"},
        {AcquisitionFunctionOption::probabilityOfImprovement, "probability-of-improvement"},
        {AcquisitionFunctionOption::expectedImprovement, "expected-improvement"},
    };
  }
 
 private:
  Value _value{Value(-1)};
};
}  // namespace options
}  // namespace autopas