drawTopicModel.py 文件源码

def drawFragmentsbyTopic(topicModel, topicIdx, n_top_frags=10, numRowsShown=4, cssTableName='fragTab', \
                         prior=-1.0, numColumns=4, tableHeader=''):    

    scores = topicModel.getTopicFragmentProbabilities()
    numTopics, numFragments = scores.shape
    if prior < 0:
        prior = 1./numFragments
    svgs=generateTopicRelatedFragmentSVGs(topicModel, topicIdx, n_top_frags=n_top_frags, prior=prior)
    namesSVGs = list(map(lambda x: "Score %.2f" % x, \
                    filter(lambda y: y > prior, sorted(scores[topicIdx,:], reverse=True)[:n_top_frags])))
    if tableHeader == '':
        tableHeader = "Topic "+str(topicIdx)
    return display(HTML(utilsDrawing.drawSVGsToHTMLGrid(svgs,tableHeader=tableHeader,cssTableName=cssTableName,\
                                                 namesSVGs=namesSVGs,size=(120,100),numRowsShown=numRowsShown,\
                                                 numColumns=numColumns)))