如何在 itertools.product() 中进行整数和元组值的参数组合？

Question

大家晚上好，我想请教一下itertools。我有在 word2vec 参数优化中进行一些手动网格搜索的代码。这是什么样子：

from gensim.models import word2vec
import itertools

parm_dict = {
    'workers' : (4),
    'size' : (300,500,750,1000),
    'min_count' : (40),
    'window' : (10)
}

workers, size, min_count, window = [tup for k,tup in parm_dict.items()] # Individual parm tuples

parm_combo = list(itertools.product(workers, size, min_count, window)) # Create all combinations

问题是，ittertools.product 无法对 worker 进行迭代，因为它被检测为整数而不是元组。它显示错误：

int type is not iteratable

请问有什么方法可以让Integer值和tuple值进行参数组合吗？或者也许还有另一种方法可以改善这一点？谢谢

Answer 1

要区分圆括号文字和元组，请添加尾随逗号

parm_dict = {
    'workers' : (4,),
    'size' : (300,500,750,1000),
    'min_count' : (40,),
    'window' : (10,)
}

如何在 itertools.product() 中进行整数和元组值的参数组合？

How to make parameter combination of integer and tuple value in itertools.product()?

python

itertools