python - 参数中的 RandomForest 分类器对象

Question

我目前正在研究随机森林分类器，使用网格搜索来获得最佳参数

所以当我得到我的参数时，它们在我的 var 中:

params = {'bootstrap': 'True', 
          'criterion': 'entropy', 
          'max_depth': 'None', 
          'max_features': '3', 
          'min_samples_leaf': '4', 
          'min_samples_split': '3'}

我想做这样的事情:

clf = RandomForestClassifier(参数)

但是这里 params 代替了 n_estimators 所以我有一些错误，比如:

ValueError: n_estimators must be an integer, got <class 'dict'>.

Answer 1

您需要解压函数调用的参数

clf = RandomForestClassifier(**params) 

让我向您展示使用字典作为具有默认参数的函数的函数参数的各种方法的结果。函数调用后的注释是打印的结果。

def foo(bar=None, baz=None):
    print(bar, baz)

params = { "bar": "Hello", "baz": "World"}

# pass params as the first parameter
foo(params) # {'baz': 'World', 'bar': 'Hello'} None

# pass keys of params as parameters (order is out of our hands)
foo(*params) # baz bar

# unpacks params with key=value as parameters
foo(**params) # Hello World

我注意到您总是在您的值中使用字符串表示形式，即使对于 None 或 True - 如果函数实际上期望 ，您可能会遇到麻烦>None 或 bool 值而不是字符串。最好检查 API 哪些值有效。