gen_adversial_model.py 文件源码

def train(sentences,discriminator_gen=discriminator_gen,discriminator=discriminator,gen_model=gen_model):

    data,real_word=getData(sentences)

    fake_word=gen_model.predict(data)
    pylab.imshow(fake_word[0][0], cmap=cm.Greens,origin='lower')

    real_label=np.zeros((1,))
    fake_label=np.ones((1,))

    #discriminator.predict(real_word)

    discriminator_gen.fit(data,fake_label,nb_epoch=5)
    fake_word=gen_model.predict(data)
    discriminator.fit(fake_word,real_label,nb_epoch=5)
    discriminator.fit(real_word,fake_label,nb_epoch=5)