訓練和測試拆分標記

Question

這是我的數據集：

import pandas as pd 
fileName= 'user.csv'
df = pd.read_csv("trainingsample_100k_apps.csv",sep=",",header=0)
pd.set_option('max_columns', None)
df

UserID  Total Usage
001       20.3
002       40.5
003       10.1

應用此查詢后，如何知道哪一行已被選為訓練和測試：

train, test = train_test_split(df, test_size = 0.20)

我希望 output 是這樣的：

UserID  Total Usage   SplitingCategory
001       20.3        Training
002       40.5        Testing
003       10.1        Training

Answer 1

好的，我得到了答案

test

它會出現我的測試數據集