BoostModel2.py 文件源码

def forward(self, char, word):
        self.models.eval()
        outs =[]
        for ii,model in enumerate(self.models):
            if model.opt.type_=='char':
                out = t.sigmoid(model(*char))
            else:
                out=t.sigmoid(model(*word))
            outs.append(out.detach())
        for ii,model in enumerate(self.new_model):
            if model.opt.type_=='char':
                out = t.sigmoid(model(*char))
            else:
                out=t.sigmoid(model(*word))
            outs.append(out)
        return sum(outs)/(len(outs))