def __init__(self, hp, model_name):
self.hp = hp
self.model_name = model_name
self.subfolder = os.getcwd() + "\\" + self.__create_subfolder()
print(self.subfolder)
self.wb1 = WeightsBias_1_0(self.hp.num_input, self.hp.num_hidden, self.hp.init_method, self.hp.eta)
self.wb1.InitializeWeights(self.subfolder, False)
self.wb2 = WeightsBias_1_0(self.hp.num_hidden, self.hp.num_output, self.hp.init_method, self.hp.eta)
self.wb2.InitializeWeights(self.subfolder, False)
def __init__(self, model_name):
self.model_name = model_name
self.subfolder = os.getcwd() + "/" + self.__create_subfolder()
self.init_method = InitialMethod.MSRA
self.eta = 0.01
# 判别器三层网络参数
self.d_wb1 = WeightsBias_1_0(784 + 10, 512, self.init_method, self.eta)
self.d_wb1.InitializeWeights(self.subfolder, True)
self.d_wb2 = WeightsBias_1_0(512, 256, self.init_method, self.eta)
self.d_wb2.InitializeWeights(self.subfolder, True)
self.d_wb3 = WeightsBias_1_0(256, 1, self.init_method, self.eta)
self.d_wb3.InitializeWeights(self.subfolder, True)
# 生成器三层网络参数
self.g_wb1 = WeightsBias_1_0(100 + 10, 256, self.init_method, self.eta)
self.g_wb1.InitializeWeights(self.subfolder, True)
self.g_wb2 = WeightsBias_1_0(256, 512, self.init_method, self.eta)
self.g_wb2.InitializeWeights(self.subfolder, True)
self.g_wb3 = WeightsBias_1_0(512, 784, self.init_method, self.eta)
self.g_wb3.InitializeWeights(self.subfolder, True)