StratifiedKFold與train_test_split分層之間的區別

Question

在訓練我的模型時，當我使用類似sklearn.model_selection.train_test_split(X, y, stratify=y, train_size=0.9)與sklearn.model_selection.StratifiedKFold(n_splits=10)之類的東西時，我得到的結果截然不同，並且想知道他們如何對數據進行分層。我幾乎可以肯定，我根據文檔實現了一切，但奇怪的是，後者給出的測試精度比第一個差得多。

Answer 1

當stratify不是無train_test_split在內部使用StratifiedShuffleSplit，而不是StratifiedKFold。所以是的，有很大的不同。