|
46 | 46 | "metrics": [{"goal": "MAXIMIZE", "metric": "accuracy"}],
|
47 | 47 | "parameters": [
|
48 | 48 | {
|
49 |
| - "integer_value_spec": {"max_value": 4, "min_value": 1}, |
50 | 49 | "parameter": "units",
|
51 | 50 | "type": "INTEGER",
|
| 51 | + "integer_value_spec": {"min_value": 1, "max_value": 4}, |
| 52 | + "scale_type": "UNIT_LINEAR_SCALE", |
52 | 53 | }
|
53 | 54 | ],
|
54 | 55 | }
|
55 | 56 | STUDY_CONFIG_INT_STEP = {
|
56 | 57 | "metrics": [{"goal": "MAXIMIZE", "metric": "accuracy"}],
|
57 | 58 | "parameters": [
|
58 | 59 | {
|
59 |
| - "discrete_value_spec": {"values": [32, 64, 96, 128]}, |
60 | 60 | "parameter": "units",
|
61 | 61 | "type": "DISCRETE",
|
| 62 | + "discrete_value_spec": {"values": [32, 64, 96, 128]}, |
62 | 63 | }
|
63 | 64 | ],
|
64 | 65 | }
|
|
69 | 70 | "double_value_spec": {"max_value": 0.5, "min_value": 0.1},
|
70 | 71 | "parameter": "learning_rate",
|
71 | 72 | "type": "DOUBLE",
|
| 73 | + "scale_type": "UNIT_LINEAR_SCALE", |
72 | 74 | }
|
73 | 75 | ],
|
74 | 76 | }
|
|
119 | 121 | "metrics": [{"goal": "MAXIMIZE", "metric": "accuracy"}],
|
120 | 122 | "parameters": [
|
121 | 123 | {
|
122 |
| - "double_value_spec": {"max_value": 1.57, "min_value": 0.0}, |
123 | 124 | "parameter": "theta",
|
124 | 125 | "type": "DOUBLE",
|
| 126 | + "double_value_spec": {"min_value": 0.0, "max_value": 1.57}, |
| 127 | + "scale_type": "UNIT_LINEAR_SCALE", |
125 | 128 | },
|
126 | 129 | {
|
127 |
| - "double_value_spec": {"max_value": 1.0, "min_value": 0.0}, |
128 | 130 | "parameter": "r",
|
129 | 131 | "type": "DOUBLE",
|
| 132 | + "double_value_spec": {"min_value": 0.0, "max_value": 1.0}, |
| 133 | + "scale_type": "UNIT_LINEAR_SCALE", |
130 | 134 | },
|
131 | 135 | ],
|
132 | 136 | }
|
@@ -245,8 +249,8 @@ def test_convert_study_config_int(self, min_value, max_value, step,
|
245 | 249 | self._assert_hps_equal(actual_hps, hps)
|
246 | 250 |
|
247 | 251 | @parameterized.parameters(
|
248 |
| - (0.1, 0.5, None, None, STUDY_CONFIG_FLOAT), |
249 |
| - (1, 2, 0.25, None, STUDY_CONFIG_FLOAT_STEP), |
| 252 | + (0.1, 0.5, None, "linear", STUDY_CONFIG_FLOAT), |
| 253 | + (1, 2, 0.25, "linear", STUDY_CONFIG_FLOAT_STEP), |
250 | 254 | (0.1, 0.8, None, "linear", STUDY_CONFIG_FLOAT_LINEAR_SCALE),
|
251 | 255 | (1e-4, 1e-1, None, "log", STUDY_CONFIG_FLOAT_LOG_SCALE),
|
252 | 256 | (1e-4, 1e-1, None, "reverse_log", STUDY_CONFIG_FLOAT_REVERSE_LOG_SCALE))
|
|
0 commit comments