9aa5a88cbc1a36f19305f229b3a9abbeb9add631,pythainlp/ulmfit/__init__.py,,document_vector,#Any#Any#Any#,84

Before Change

    m = learn.model[0]
    m.reset()
    pred,_ = m(t)
    res = pred[-1][-1,:,:].squeeze().detach().numpy()
    return(res)

After Change

    pred,_ = m(t)
    //return concatenation of last, mean and max
    last = pred[-1][-1,:,:].squeeze()
    avg_pool = pred[-1].mean(0)[0].squeeze()
    max_pool = pred[-1].max(0)[0].squeeze()
    res = torch.cat((last,avg_pool,max_pool)).detach().cpu().numpy()
    return(res)

Instances