Python TFIDFの例

コード例 #1

ファイル: test_tf_idf.py プロジェクト: Kensuke-Mitsuzawa/DocumentFeatureSelection

    def test_output_result_pmi(self):
        tf_idf_weighted_matrix = tf_idf.TFIDF().fit_transform(
            X=self.csr_matrix_,
        )
        assert isinstance(tf_idf_weighted_matrix, csr_matrix)

        tf_idf_scored_dict = ScoredResultObject(
            scored_matrix=tf_idf_weighted_matrix,
            label2id_dict=self.label2id_dict,
            feature2id_dict=self.vocabulary,
        ).convert_score_matrix2score_record(outformat='items')
        self.assertTrue(isinstance(tf_idf_scored_dict, list))
        assert isinstance(tf_idf_scored_dict, list)

コード例 #2

ファイル: test_tf_idf.py プロジェクト: Python3pkg/DocumentFeatureSelection

    def test_output_result_pmi(self):
        import numpy

        tf_idf_weighted_matrix = tf_idf.TFIDF().fit_transform(
            X=self.csr_matrix_, )
        assert isinstance(tf_idf_weighted_matrix, csr_matrix)

        tf_idf_scored_dict = ScoredResultObject(
            scored_matrix=tf_idf_weighted_matrix,
            label2id_dict=self.label2id_dict,
            feature2id_dict=self.vocabulary,
        ).ScoreMatrix2ScoreDictionary(outformat='items', n_jobs=-1)

        assert isinstance(tf_idf_scored_dict, list)
        import pprint
        pprint.pprint(tf_idf_scored_dict)

コード例 #3

ファイル: test_tf_idf.py プロジェクト: Kensuke-Mitsuzawa/DocumentFeatureSelection

 def test_normal_fit_transform(self):
     tf_idf_weighted_matrix = tf_idf.TFIDF().fit_transform(
         X=self.csr_matrix_,
     )
     assert isinstance(tf_idf_weighted_matrix, csr_matrix)