使用 python 的sklearn模块,训练模型时,出现 ValueError: min_samples_split must be at least 2 or in (0, 1], got 1 的

来源:互联网 发布:壁纸桌面软件下载 编辑:程序博客网 时间:2024/06/07 12:09

最近使用python sklearn模块中的 sklearn.ensembles.RandomForestRegressor,训练随机森林模型,出现错误: ValueError: min_samples_split must be at least 2 or in (0, 1], got 1 

并且据我所知,其他模型训练模块也会出现同样问题。


代码段为:



regr = RandomForestRegressor(n_estimators=50,max_depth=None,min_samples_split=1,random_state=0)

regr = regr.fit(features_train,ratings_train)


解决方法是: 将第一行中的min_samples_split=1 改为 min_samples_split=1.0



阅读全文
0 0
原创粉丝点击