import pickle


def load_digits_data(filename='./dataset/digits-dataset'):
  """
  Load the dataset in training and testing sets
  """
  with open(filename, 'rb') as f:
    x_train, y_train, x_test, y_test = pickle.load(f)
  return x_train, y_train, x_test, y_test