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)
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)
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)