【发布时间】:2018-09-03 11:10:10
【问题描述】:
我用下面的代码做了随机森林回归和交叉验证。
我有两个问题。
- 我不明白案例中交叉验证中的“分数”是什么意思 回归,而不是分类(我的意思是在默认情况下)。什么 这是?
- 还有设置其他功能的功能(如 mean_squared_error) 作为分数?
谢谢。
from sklearn.ensemble import RandomForestRegressor
rf = RandomForestRegressor(n_estimators=100, criterion='mse',random_state=1, n_jobs=-1)
rf.fit(X_trainr, y_trainr)
y_train_pred = rf.predict(X_trainr)
y_test_pred = rf.predict(X_testr)
y_all_pred = rf.predict(xsetr)
from sklearn.model_selection import cross_val_score
scores = cross_val_score(rf, xsetr, ysetr, cv=5)
print("Cross-validation scores:{}".format(scores))
【问题讨论】:
标签: machine-learning random-forest cross-validation