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)