generator.py 文件源码

def update_sequence_weights(self,pids,Alpha,features,sequences,publish_years,predict_year,beta,W1,W2):

        result = []
        for i in range(len(pids)):
            seq = {}
            fea = numpy.mat(features[i])
            beta = numpy.mat(beta)
            seq['seq_id'] = i
            seq['paper_id'] = pids[i]
            seq['theta'] = W1[i]
            seq['w'] = W2[i]
            seq['alpha'] = Alpha[i]
            seq['fea'] = features[i]
            seq['beta'] = beta.tolist()
            seq['spont'] = (fea*beta).tolist()[0]
            result.append(seq)

        self.sequence_weights = result