词嵌入层的对抗训练组件
- 增加模型鲁棒性和泛化性
pip install adversal_embedding
from keras.models import load_model
from adversal_embedding.core import adversarial_training
model = load_model(model_dir)
model.compile(optimizer=optimizer, loss=loss, metrics=metrics)
adversarial_training(model, 'embedding_vocab', 0.5) # 用在compile之后
model.fit(train_data, test_data, epochs=epochs, batch_size=batch_size)