numpy_serialize.py

with open('intermediate/pre_processed_data.bin', 'wb') as f:
  np.save(f, X_train)
  np.save(f, y_train)
  np.save(f, X_val)
  np.save(f, y_val)
  np.save(f, X_test)
  np.save(f, y_test)


with open('intermediate/pre_processed_data.bin', 'rb') as f:
  X_train = np.load(f)
  y_train = np.load(f)
  X_val = np.load(f)
  y_val = np.load(f)
  X_test = np.load(f)
  y_test = np.load(f)