【发布时间】:2019-08-21 08:03:51
【问题描述】:
我是 scikit learn 的新手,所以请原谅我的无知。使用 GridsearchCV 我正在尝试优化 DecisionTreeRegressor。我使参数空间越宽,得分越差。
将 min_samples_split 设置为 range(2,10) 会得到 -0.04 的 neg_mean_squared_error。设置为 range(2,5) 时得分为-0.004。
simple_tree =GridSearchCV(tree.DecisionTreeRegressor(random_state=42), n_jobs=4, param_grid={'min_samples_split': range(2, 10)}, scoring='neg_mean_squared_error', cv=10, refit='neg_mean_squared_error')
simple_tree.fit(x_tr,y_tr).score(x_tr,y_tr)
与不那么广泛的网格搜索相比,我希望更广泛的网格搜索获得相等或更高的正分数。
【问题讨论】:
标签: scikit-learn decision-tree gridsearchcv