结合所有参数进行网格搜索

时间:2019-02-25 03:47:59

标签: python python-3.x grid-search

我正在训练FastText单词表示模型,并且正在对多个参数范围进行网格搜索。以下是多个参数列表:

wordNgrams = [2, 3, 4, 5]
lr = [10e-2, 10e-3, 10e-4, 10e-5, 10e-6]
dim = [200, 250, 300]
ws = [5, 6, 8, 10]

我想将上述列表的所有可能组合尝试在一起,并将它们作为参数传递以训练我的模型。我不知道如何实现python函数来做到这一点,并希望获得一些帮助。请帮忙。

2 个答案:

答案 0 :(得分:1)

IIUC使用i=0 for key,val in inp.items(): if i<3: print(f"{inp['output['+str(i)+'][description]']}:{inp['output['+str(i)+'][input]']}") i+=1

itertools

答案 1 :(得分:0)

您也可以使用 sklearn.model_selection 中的 ParameterGrid。在回答类似问题 here 时解释得更好。