Numpy 2d Array Size Python Compiler With Sklearn Train Test Split