hpseq_baselines.py 文件源码

def evaluate_stance(
        self,
        testN,
        testtimes,
        testinfecting_vec,
        testinfected_vec,
        testeventmemes,
        testW,
        testT,
        ):
        predictednode_vec = []

        for next_event_index in xrange(len(testtimes)):
            print 'considering event', next_event_index
            words = testW[next_event_index, :]

            predictions = []
            for label in set(self.node_vec):
                loglikelihood_term = 0
                loglikelihood_term += np.dot(words,
                        np.log(self.beta[label, :]))
                predictions.append((label, loglikelihood_term))
            predictednode_vec.append(max(predictions, key=lambda x: \
                    x[1])[0])
        return predictednode_vec