fix hyperopt quniform/qloguniform issue

4feb466c · suiguoxin · 591fa1d7 · 4feb466c
Commit 4feb466c authored Aug 04, 2019 by suiguoxin
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py +4 -1

No files found.
--- a/src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py
+++ b/src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py
@@ -90,6 +90,9 @@ def json2parameter(in_x, parameter, name=NodeType.ROOT):
                                   parameter,
                                   name=name + '[%d]' % _index)
                }
+            else:
+                if _type in ['quniform', 'qloguniform']:
+                    out_y = np.clip(parameter[name], in_x[NodeType.VALUE][0], in_x[NodeType.VALUE][1])
                else:
                    out_y = parameter[name]
        else: