def test_convert_optimizer_trial_to_hps(self):
hps = hp_module.HyperParameters()
hps.Choice("learning_rate", [1e-4, 1e-3, 1e-2])
optimizer_trial = {
"name":
"trial_name",
"state":
"ACTIVE",
"parameters": [
{
"parameter": "learning_rate",
"floatValue": 0.0001
},
{
"parameter": "num_layers",
"intValue": "2"
},
{
"parameter": "units_0",
"floatValue": 96
},
{
"parameter": "units_1",
"floatValue": 352
},
],
}
trial_hps = utils.convert_optimizer_trial_to_hps(hps, optimizer_trial)
self.assertEqual(trial_hps.values, EXPECTE_TRIAL_HPS)
def test_convert_optimizer_trial_to_hps(self):
hps = hp_module.HyperParameters()
hps.Choice("learning_rate", [1e-4, 1e-3, 1e-2])
trial_hps = utils.convert_optimizer_trial_to_hps(hps, OPTIMIZER_TRIAL)
self.assertDictEqual(trial_hps.values, EXPECTED_TRIAL_HPS)