Я пытаюсь установить столбцы, используемые для страт, в стратифицированную складку, но не нашел, как это сделать. В train_test_split мы можем использовать stratify = (любой столбец / подмножество, используемое для стратификации), но неудобно запускать train_test_split 5/10 раз. У кого-нибудь есть идея сделать это?