GroupKFold-小記

GroupKFold是KFold一個變體,目的在於將groups嚴格分開,以減少過擬合現象。
它如何work的呢?
看下圖,group參數是通過不同的group標識符(用顏色來區分),使得訓練集和測試集在劃分時不會跨group的範圍。不會overlap。
比如以下代碼。表明有三個組,即三個不同顏色劃分。

>>> groups = [1, 1, 1, 2, 2, 2, 3, 3, 3, 3]

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章