Python SimpleTermDocumentMatrix.get_terms Examples

Example #1

File: priority_index.py Project: kentan/Machine-Learning-For-Hackers-in-Python

    def get_message_weight(self,body):
        body_tdm = SimpleTermDocumentMatrix()
        body_tdm.add_doc(body)
        body_terms = body_tdm.get_terms(min_doc_freq=1)

        tm = self.tdm.get_term_freq()

        for body_term in body_terms:
            tm_log10 = [math.log10(tm[k]) for k in tm if body_term == k]

        mean =  sum(tm_log10)/len(tm_log10) if len(tm_log10) > 0 else 1
        return mean