我需要從一個設定的日期創建測驗和訓練。但是，我不能完全用 sklears 做到這一點。
My Target variable: SalePrice
train = pd.read_csv(r'C:\Users\pkoni\Desktop\train.csv')
target = train['SalePrice']
X, y = train.data, train.target
train_X, test_X, train_y, test_y = train_test_split(X, y,
train_size=0.5,
test_size=0.5,
random_state=123)
我不知道我應該在 X, y 中添加什么。
uj5u.com熱心網友回復：
不確定我是否完全理解。如果您只是嘗試隨機拆分，那么這應該可行：
y = train['SalePrice']
X = train.drop('SalePrice', axis=1)
X_train, X_test, y_train, y_test = train_test_split(X, y,
test_size=0.5,
random_state=0)
如果您想在某個日期（例如 2010 年）之后拆分所有點進行測驗，并在訓練之前拆分所有點，則需要不同的解決方案。
test = train[train['Yr.Sold'] < 2010]
train = train[train['Yr.Sold'] > 2010]
然后在拆分測驗和訓練之后，您可以為每個分配標簽和特征（參見第一個代碼段中的 x,y）。
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/476503.html