def __init__(self, max_iter=10):
train_imgs, train_labels, test_imgs, test_labels = dataset_loader.load(
'mnist')
train_imgs = train_imgs.reshape([-1, 784])
test_imgs = test_imgs.reshape([-1, 784])
self.X_train = train_imgs
self.Y_train = train_labels
self.X_test = test_imgs
self.Y_test = test_labels
self.max_iter = max_iter
self.cur_iter = 0
def __init__(self):
train_imgs, train_labels, test_imgs, test_labels = dataset_loader.load(
'fashionmnist')
train_imgs = train_imgs.reshape([-1, 784])
test_imgs = test_imgs.reshape([-1, 784])
self.X_train = train_imgs
self.X_test = test_imgs
self.train_label = train_labels
self.test_label = test_labels
self.sets = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
self.max_iter = len(self.sets)
self.cur_iter = 0
def __init__(self):
train_imgs, train_labels, test_imgs, test_labels = dataset_loader.load(
'mnist')
train_imgs = train_imgs.reshape([-1, 784])
test_imgs = test_imgs.reshape([-1, 784])
self.X_train = train_imgs
self.X_test = test_imgs
self.train_label = train_labels
self.test_label = test_labels
self.sets_0 = [0, 2, 4, 6, 8]
self.sets_1 = [1, 3, 5, 7, 9]
self.max_iter = len(self.sets_0)
self.cur_iter = 0
def __init__(self):
train_imgs, train_Y, test_imgs, test_Y = dataset_loader.load(
'fashionmnist')
train_X = train_imgs.reshape([-1, 784])
test_X = test_imgs.reshape([-1, 784])
self.X_train = train_X.reshape([-1, 784]) / 255
self.X_test = test_X.reshape([-1, 784]) / 255
self.train_label = train_Y
self.test_label = test_Y
self.sets_0 = [0, 2, 4, 6, 8]
self.sets_1 = [1, 3, 5, 7, 9]
self.max_iter = len(self.sets_0)
self.cur_iter = 0