(self, opt='adam')
| 105 | (self.samples_per_epoch - 1) // self.batch_size + 1) * times |
| 106 | |
| 107 | def reset_model(self, opt='adam'): |
| 108 | |
| 109 | self.model, self.embedding_dict = create_model( |
| 110 | self.node_size, self.rep_size, self.order) |
| 111 | self.model.compile(opt, line_loss) |
| 112 | self.batch_it = self.batch_iter(self.node2idx) |
| 113 | |
| 114 | def _gen_sampling_table(self): |
| 115 |
no test coverage detected