def test_ei_2obj(self):
ei = MultiObjExpectedImprovement()
ei.target = array([[1, 10], [1, -10]])
ei.current = [NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=0, sigma=1)]
ei.calc_switch = "EI"
ei.execute()
self.assertAlmostEqual([5.0], ei.EI,1)
self.assertEqual(0.5, ei.PI,6)
def test_ei_2obj(self):
ei = MultiObjExpectedImprovement()
ei.target = array([[1, 10], [1, -10]])
ei.current = [
NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=0, sigma=1)
]
ei.calc_switch = "EI"
ei.execute()
self.assertAlmostEqual([5.0], ei.EI, 1)
self.assertEqual(0.5, ei.PI, 6)
def test_ei_calc_switch(self):
ei = MultiObjExpectedImprovement()
ei.target = array([[1, 1, 1]])
ei.current = [NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=1, sigma=1)]
ei.calc_switch = 'EI'
try:
ei.execute()
except ValueError,err:
self.assertEqual(str(err),': EI calculations not supported'
' for more than 2 objectives')
def test_ei_2obj(self):
ei = MultiObjExpectedImprovement()
bests = CaseSet()
list_of_cases = [Case(outputs=[("y1", 1), ("y2", 10)]), Case(outputs=[("y1", 1), ("y2", -10)])]
for case in list_of_cases:
bests.record(case)
ei.best_cases = bests
ei.criteria = ["y1", "y2"]
ei.predicted_values = [NormalDistribution(mu=1, sigma=1), NormalDistribution(mu=0, sigma=1)]
ei.calc_switch = "EI"
ei.execute()
self.assertAlmostEqual([5.0], ei.EI, 1)
self.assertEqual(0.5, ei.PI, 6)
def test_ei_2obj(self):
ei = MultiObjExpectedImprovement()
bests = CaseSet()
list_of_cases = [Case(outputs=[("y1",1),("y2",10)]),Case(outputs=[("y1",1),("y2",-10)])]
for case in list_of_cases:
bests.record(case)
ei.best_cases = bests
ei.criteria = ["y1","y2"]
ei.predicted_values = [NormalDistribution(mu=1,sigma=1),NormalDistribution(mu=0,sigma=1)]
ei.calc_switch = "EI"
ei.execute()
self.assertAlmostEqual([5.0],ei.EI,1)
self.assertEqual(0.5,ei.PI,6)
def test_ei_calc_switch(self):
ei = MultiObjExpectedImprovement()
ei.target = array([[1, 1, 1]])
ei.current = [
NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=1, sigma=1),
NormalDistribution(mu=1, sigma=1)
]
ei.calc_switch = 'EI'
try:
ei.execute()
except ValueError, err:
self.assertEqual(
str(err), ': EI calculations not supported'
' for more than 2 objectives')
def test_ei_calc_switch(self):
ei = MultiObjExpectedImprovement(3)
bests = CaseSet()
list_of_cases = [Case(outputs=[("y1",1),("y2",1),("y3",1)])]
for case in list_of_cases:
bests.record(case)
ei.best_cases = bests
ei.criteria = ['y1','y2','y3']
ei.predicted_values = [NormalDistribution(mu=1,sigma=1),
NormalDistribution(mu=1,sigma=1),
NormalDistribution(mu=1,sigma=1)]
ei.calc_switch = 'EI'
try:
ei.execute()
except ValueError,err:
self.assertEqual(str(err),': EI calculations not supported'
' for more than 2 objectives')
