Cat Faces dataset preprocessing function

cat_utils.py

import numpy as np
from sklearn.model_selection import train_test_split

def load_cats():
    cats = np.load("./cats.npy")
    print (cats.shape)
    Y = []
    for i in range(cats.shape[0]):
        Y.append([1,0])
    Y = np.array(Y)

    (x_train, y_train, x_test, y_test) = train_test_split(cats, Y)
    x_train = (x_train.astype(np.float32)) / 255
    x_train = x_train.reshape(x_train.shape[0], 784)
    
    return (x_train, y_train, x_test, y_test)