Skip to content

GitLab

Commit 4feb466c authored by suiguoxin's avatar suiguoxin
Browse files

fix hyperopt quniform/qloguniform issue

parent 591fa1d7
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
src/sdk/pynni/nni/hyperopt_tuner/hyperopt_tuner.py
View file @ 4feb466c
...@@ -90,6 +90,9 @@ def json2parameter(in_x, parameter, name=NodeType.ROOT): ...@@ -90,6 +90,9 @@ def json2parameter(in_x, parameter, name=NodeType.ROOT):
parameter, parameter,
name=name + '[%d]' % _index) 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: else:
out_y = parameter[name] out_y = parameter[name]
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment