| 22 | class DygraphModel(): |
| 23 | # define model |
| 24 | def create_model(self, config): |
| 25 | max_sentence = config.get('hyper_parameters.max_sentence', 50) |
| 26 | max_sents = config.get('hyper_parameters.max_sents', 30) |
| 27 | max_entity_num = config.get('hyper_parameters.max_entity_num', 10) |
| 28 | npratio = config.get('hyper_parameters.npratio', 4) |
| 29 | hidden_size = config.get('hyper_parameters.hidden_size', 400) |
| 30 | embedding_size = config.get('hyper_parameters.embedding_size', 300) |
| 31 | vocab_size = config.get('hyper_parameters.vocab_size', 3030) |
| 32 | kim_model = net.KIMLayer( |
| 33 | vocab_size, |
| 34 | embedding_size, |
| 35 | hidden_size, |
| 36 | max_sents, |
| 37 | max_sentence, |
| 38 | max_entity_num, ) |
| 39 | |
| 40 | return kim_model |
| 41 | |
| 42 | # define feeds which convert numpy of batch data to paddle.tensor |
| 43 | def create_feeds(self, batch_data): |