nn1_stress_test.py 文件源码

def __init__(self, batch_size, word_gru_hidden, feature_dim, n_classes, bidirectional=True):        

        super(MixtureSoftmax, self).__init__()

        # for feature only model 
        word_gru_hidden = 0
        # end

        self.batch_size = batch_size
        self.n_classes = n_classes
        self.word_gru_hidden = word_gru_hidden
        self.feature_dim = feature_dim

        if bidirectional == True:
            self.linear = nn.Linear(2 * 2 * word_gru_hidden + feature_dim, n_classes)
        else:
            self.linear = nn.Linear(2 * word_gru_hidden + feature_dim, n_classes)