所以我得到了一個包含 120 行和 124 列的數據集。從屬列是人的字符串名稱?？偣灿?0個不同的名字。我想從每個類中提取兩行（因此為每個類提取兩行具有相同名稱的行）以便我可以用它創建一個測試集。任何幫助將不勝感激。謝謝
1 回答
三國紛爭
TA貢獻1804條經驗 獲得超7個贊
您可以使用以下函數來執行此操作：
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2,stratify =y)
這里，
X = 包含所有自變量的數據框。
y = 具有因變量的序列。
test_size = 你想要作為測試大小的百分比，這里是 20%。
stratify = 所有類別的平均分配，在你的例子中是 20 個人的名字。
添加回答
舉報
0/150
提交
取消