def METHOD_NAME(cls):
    (x_train, y_train), (x_test, y_test), _, _ = load_dataset("mnist")
    cls.n_classes = 10
    cls.n_features = 28 * 28
    n_train = x_train.shape[0]
    n_test = x_test.shape[0]
    x_train = x_train.reshape((n_train, cls.n_features))
    x_test = x_test.reshape((n_test, cls.n_features))
    cls.x_train = x_train[:NB_TRAIN]
    cls.y_train = y_train[:NB_TRAIN]
    cls.x_test = x_test[:NB_TEST]
    cls.y_test = y_test[:NB_TEST]