我正在尝试训练多元线性回归模型。我有一个名为“main”的数据集。该数据集中的分类变量很少。我对分类变量进行了虚拟化。假设虚拟化后得到的列是A、B、C、D等。现在，当我尝试在这个主数据集上运行训练测试拆分时，由此获得的训练数据集在这些列之一中只有值 0。我该如何克服这个问题。
我正在使用的代码是：
对于训练测试拆分：
from sklearn.model_selection import train_test_split
np.random.seed(0)
df_train, df_test = train_test_split(main, train_size = 0.7, test_size = 0.3, random_state = 100)
在运行以下代码时：
main.columns[main.nunique() == 1]
结果是：Index([], dtype='object')
并且在为火车数据运行以下代码时：
df_train.columns[df_train.nunique() == 1]
结果是：Index(['A', 'D', 'S'], dtype='object')
我希望生成的火车集包含具有所有值组合的特征。但是，这种拆分在某些功能中只给了我一个值
编辑：我检查了这些列中的唯一值，这些列高度不平衡，只有一个值出现在正例中。我尝试分层，它至少需要两排正类。许多列都是这种情况。所以我不能在训练数据集中单独包含这些列，因为它需要为所有列编写代码。我希望这能自动完成。