我有一本名為 params 的字典。我需要遍歷這個字典,以便它按順序列印 learn_rate 和 btch_size 的所有可能組合。這是我的字典:
params = dict(
learn_rate = [0.01, 0.001, 0.0001]
,btch_size = [50, 70, 80]
)
所需的輸出:
0.01 50
0.01 70
0.01 80
0.001 50
0.001 70
0.001 80
0.0001 50
0.0001 70
0.0001 80
我正在考慮為此使用 itertools,因為我不想要大量嵌套的 for 回圈,以防我決定添加新串列。有沒有人有什么建議?
uj5u.com熱心網友回復:
'itertools.product()' 正是這樣做的。適用于您要添加的任意數量的串列。
>>> list(itertools.product(*params.values()))
[(0.01, 50), (0.01, 70), (0.01, 80), (0.001, 50), (0.001, 70), (0.001, 80), (0.0001, 50), (0.0001, 70), (0.0001, 80)]
>>>
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/453325.html