Train_test_split error

#1

I tried using the code:
XT,Xt,YT,Yt=train_test_split(X, Y, test_size=.2, random_state=5)
to split the train and test data but it shows an error:
MemoryError: Unable to allocate array with shape (33600, 784) and data type int64

#2

Screenshot:

SS
SS1530×1028 179 KB

#3

Hey @SanchitSayala, this is due to ram overflow error, switch to google colab, they will provide you 12 gb of free ram.

Hope this resolved your doubt.
Plz mark the doubt as resolved in my doubts section. :blush:

#4

It worked perfectly after restarting the kernel. Thanks anyways. :+1: