我是Python的新手,在创建函数时遇到了一些问题:
我创建了一个字典,其中的键是参数,例如n_estimators
、C
、max_depths
。
我有一个循环,想为给定的估计器(这是我的函数的输入)设置参数,但遇到了问题。
例如,假设我的估计器是RandomForestClassifier
,
代码将是:
key = 'n_estimators'estimator = estimator.set_params(key=100)
我得到了以下错误:
ValueError: Invalid parameter key for estimator RandomForestClassifier.
我明白问题所在,set_params
将key
视为参数(而不是n_estimators
),但我还不知道如何解决这个问题。
我非常希望能在这里得到一些建议。
回答:
使用带有**
运算符的字典:
key = 'n_estimators'estimator = estimator.set_params(**{key: 100})