continue_setup = Setup('')
continue_setup.load(rel_filepath=rel_filepath)
change_lr = None
if change_lr is not None:
K.set_value(continue_setup.getModel().optimizer.lr, change_lr)
print('Changing the model optimizer learning rate to = %f' % K.get_value(continue_setup.getModel().optimizer.lr))
else:
print('Model optimizer learning rate = %f' % K.get_value(continue_setup.getModel().optimizer.lr))
X_train_cnn, y_train_one_hot, X_val_cnn, y_val_one_hot, X_test_cnn, y_test_one_hot = continue_setup.getData()
for epoch in range(continue_setup.getEpoch() + 1, 10000):
print('Training \'%s\': Epoch %d' % (continue_setup.getName(), epoch))
dropout = continue_setup.getModel().fit(X_train_cnn, y_train_one_hot,
batch_size=64, epochs=1, verbose=1,
validation_data=(X_val_cnn, y_val_one_hot))
continue_setup.updateEpochs(add_epochs=1,
train_acc=dropout.history['acc'],
train_loss=dropout.history['loss'],
val_acc=dropout.history['val_acc'],
val_loss=dropout.history['val_loss'],
test_acc=[0],
test_loss=[0],
allow_modify=True)
continue_setup.save('setup')
class TestSetup(TestCase):
@classmethod
def setUpClass(cls):
(cls.train_X, cls.train_Y), (cls.test_X, cls.test_Y) = fashion_mnist.load_data()
(cls.train_X, cls.train_Y), (cls.test_X, cls.test_Y) = (cls.train_X[:100], cls.train_Y[:100]), (cls.test_X[:100], cls.test_Y[:100])
cls.train_X = cls.train_X.reshape(-1, 28, 28, 1)
cls.test_X = cls.test_X.reshape(-1, 28, 28, 1)
cls.train_X = cls.train_X.astype('float32')
cls.test_X = cls.test_X.astype('float32')
cls.train_X = cls.train_X / 255.
cls.test_X = cls.test_X / 255.
# Change the labels from categorical to one-hot encoding
train_Y_one_hot = to_categorical(cls.train_Y)
test_Y_one_hot = to_categorical(cls.test_Y)
cls.train_X, cls.valid_X, train_label, cls.valid_label = train_test_split(cls.train_X, train_Y_one_hot, test_size=0.2,
random_state=13)
batch_size = 64
epochs = 2
num_classes = 10
cls.fashion_model = Sequential()
cls.fashion_model.add(Conv2D(32, kernel_size=(3, 3), activation='linear', input_shape=(28, 28, 1), padding='same'))
cls.fashion_model.add(LeakyReLU(alpha=0.1))
cls.fashion_model.add(MaxPooling2D((2, 2), padding='same'))
cls.fashion_model.add(Conv2D(64, (3, 3), activation='linear', padding='same'))
cls.fashion_model.add(LeakyReLU(alpha=0.1))
cls.fashion_model.add(MaxPooling2D(pool_size=(2, 2), padding='same'))
cls.fashion_model.add(Conv2D(128, (3, 3), activation='linear', padding='same'))
cls.fashion_model.add(LeakyReLU(alpha=0.1))
cls.fashion_model.add(MaxPooling2D(pool_size=(2, 2), padding='same'))
cls.fashion_model.add(Flatten())
cls.fashion_model.add(Dense(128, activation='linear'))
cls.fashion_model.add(LeakyReLU(alpha=0.1))
cls.fashion_model.add(Dense(num_classes, activation='softmax'))
cls.fashion_model.compile(loss=keras.losses.categorical_crossentropy, optimizer=keras.optimizers.Adam(),
metrics=['accuracy'])
cls.fashion_train = cls.fashion_model.fit(cls.train_X, train_label, batch_size=batch_size, epochs=epochs, verbose=1,
validation_data=(cls.valid_X, cls.valid_label))
cls.test_eval = cls.fashion_model.evaluate(cls.test_X, test_Y_one_hot, verbose=0)
def setUp(self):
self.setup = Setup('fashion_model')
def test_updateEpochs_correctList_noModify(self):
mList = [1, 2, 3]
self.setup.updateEpochs(3, mList, mList, mList, mList, mList, mList, allow_modify=False)
self.assertEqual(self.setup._train_accuracy, [1, 2, 3])
self.assertEqual(self.setup._train_loss, [1, 2, 3])
self.assertEqual(self.setup._val_accuracy, [1, 2, 3])
self.assertEqual(self.setup._val_loss, [1, 2, 3])
self.assertEqual(self.setup._test_accuracy, [1, 2, 3])
self.assertEqual(self.setup._test_loss, [1, 2, 3])
def test_updateEpochs_smallerList_noModify(self):
mList = [1, 2, 3]
self.assertRaises(ValueError, self.setup.updateEpochs, 3, mList, mList, mList[:-1], mList, mList, mList, allow_modify=False)
def test_updateEpochs_largerList_noModify(self):
mList = [1, 2, 3]
self.assertRaises(ValueError, self.setup.updateEpochs, 3, mList, mList, mList, mList, mList, mList + [4], allow_modify=False)
def test_updateEpochs_correctList_modifyAllowed(self):
mList = [1, 2, 3]
self.setup.updateEpochs(3, mList, mList, mList, mList, mList, mList, allow_modify=True)
self.assertEqual(self.setup._train_accuracy, [1, 2, 3])
self.assertEqual(self.setup._train_loss, [1, 2, 3])
self.assertEqual(self.setup._val_accuracy, [1, 2, 3])
self.assertEqual(self.setup._val_loss, [1, 2, 3])
self.assertEqual(self.setup._test_accuracy, [1, 2, 3])
self.assertEqual(self.setup._test_loss, [1, 2, 3])
def test_updateEpochs_smallerList_modifyAllowed(self):
mList = [1, 2, 3]
self.setup.updateEpochs(3, mList, mList, mList, mList, mList[:-1], mList, allow_modify=True)
self.assertEqual(self.setup._train_accuracy, [1, 2, 3])
self.assertEqual(self.setup._train_loss, [1, 2, 3])
self.assertEqual(self.setup._val_accuracy, [1, 2, 3])
self.assertEqual(self.setup._val_loss, [1, 2, 3])
self.assertEqual(self.setup._test_accuracy, [1, 2, 2])
self.assertEqual(self.setup._test_loss, [1, 2, 3])
def test_updateEpochs_largerList_modifyAllowed(self):
mList = [1, 2, 3]
self.setup.updateEpochs(3, mList, mList, mList, mList, mList, mList + [4], allow_modify=True)
self.assertEqual(self.setup._train_accuracy, [1, 2, 3])
self.assertEqual(self.setup._train_loss, [1, 2, 3])
self.assertEqual(self.setup._val_accuracy, [1, 2, 3])
self.assertEqual(self.setup._val_loss, [1, 2, 3])
self.assertEqual(self.setup._test_accuracy, [1, 2, 3])
self.assertEqual(self.setup._test_loss, [1, 2, 3])
def test_positive(self):
self.setup.setModel(self.fashion_model)
self.setup.setData(XTrain=self.train_X)
self.setup.setData(XValidation=self.valid_X)
self.setup.setData(XTest=self.test_X)
self.setup.setData(YTrain=self.train_Y)
self.setup.setData(YValidation=self.valid_label)
self.setup.setData(YTest=self.test_Y)
self.setup.save('setup')