官术网_书友最值得收藏!

K-fold cross-validation

This method was invented and gained popularity in those days when the big date was not yet a problem, everyone had little data, but still needed to build reliable models. First thing we do is shuffle our dataset well, and then divide it randomly into several equal parts, say 10 (this is the k in k-fold). We hold out the first part as a test set, and on the remaining nine parts we train the model. The trained model is then assessed on the test set that did not participate in the training as usual. Next, we hold out the second of 10 parts, and train the model on the remaining nine (including those previously served as a test set). We validate the new model again on the part that did not participate in the training. We continue this process until each of the 10 parts is in the role of the test set. The final quality metrics are determined by the averaging metrics from each of the 10 tests:

In []: 
from sklearn.model_selection import cross_val_score 
scores = cross_val_score(tree_model, features, df.label, cv=10) 
np.mean(scores) 
Out[]: 
0.88300000000000001 
In []: 
plot = plt.bar(range(1,11), scores) 
Out[]: 
Figure 2.10: Cross-validation results

From the preceding graph, you can see that the model's accuracy depends on how you split the data, but not much. By taking the average and variance of the cross-validation results, you can make a sense of how well your model can generalize on different data, and how stable it is.

主站蜘蛛池模板: 惠东县| 图片| 郎溪县| 乌恰县| 广宁县| 北京市| 嘉善县| 东兰县| 东宁县| 云安县| 贡山| 广昌县| 邵东县| 中卫市| 新乡县| 郁南县| 清镇市| 泾阳县| 富宁县| 开阳县| 洪江市| 丹棱县| 渭南市| 乐平市| 淮安市| 江山市| 卓资县| 凤凰县| 湖北省| 将乐县| 沁水县| 墨玉县| 广饶县| 安新县| 神木县| 潞城市| 曲水县| 海南省| 海安县| 施秉县| 张北县|