AI代码类
wateryh
这个作者很懒,什么都没留下…
展开
-
torch初始化
1. 线性torch.nn.init.xavier_normal_(self.liner_full_1.weight)torch.nn.init.xavier_normal_(self.liner_full_2.weight)self.word_embeds = nn.Embedding(vocab_size, embedding_dim)pretrained_weight = np.array(pretrained_weight)self.embed.weight.data.copy_(原创 2021-02-25 22:16:26 · 618 阅读 · 0 评论 -
torch模型保存代码
class myBertMode(torch.nn.Module): def save_model(self,mode_path): torch.save(self.state_dict(), mode_path) def load_model(self,mode_path): checkpoint = torch.load(mode_path) self.load_state_dict(checkpoint)...原创 2021-02-24 22:07:07 · 197 阅读 · 0 评论 -
停用词
结巴分词后的停用词性 [标点符号、连词、助词、副词、介词、时语素、‘的’、数词、方位词、代词]stop_flag = ['x', 'c', 'u','d', 'p', 't', 'uj', 'm', 'f', 'r']原创 2021-01-28 06:39:23 · 840 阅读 · 0 评论