financial_insights.py 文件源码

def rank_bank_topic(bank_doc_map, doc_topic_dist):
    """Rank topics for banks
    """
    bank_topic_ranks = {}
    for each_bank in bank_doc_map:
        rank = []
        for each_doc in bank_doc_map[each_bank]:
            rank.append(calc_ranks(doc_topic_dist[each_doc]))
        rank = np.r_[rank]
        # compute ranking score
        bank_topic_ranks[each_bank] = np.mean(1. / rank, axis=0)
    return bank_topic_ranks