ranpelta/split the data to train and test.py

## split the data to train and test.py
test_size = int(len(df) * 0.1) # the test data will be 10% (0.1) of the entire data
train = df.iloc[:-test_size,:].copy()
# the copy() here is important, it will prevent us from getting: SettingWithCopyWarning: A value is trying to be set on a copy of a slice from a DataFrame. Try using .loc[row_index,col_indexer] = value instead
test = df.iloc[-test_size:,:].copy()
print(train.shape, test.shape)

>>> ((28916, 4), (3212, 4))
	test_size = int(len(df) * 0.1) # the test data will be 10% (0.1) of the entire data
	train = df.iloc[:-test_size,:].copy()
	# the copy() here is important, it will prevent us from getting: SettingWithCopyWarning: A value is trying to be set on a copy of a slice from a DataFrame. Try using .loc[row_index,col_indexer] = value instead
	test = df.iloc[-test_size:,:].copy()
	print(train.shape, test.shape)

	>>> ((28916, 4), (3212, 4))