Skip to content

Instantly share code, notes, and snippets.

@ZhangzihanGit

ZhangzihanGit/select_words.py

Created May 16, 2022 10:22
Show Gist options
  • Save ZhangzihanGit/3c04ffc30b141960f5162e7ffd348d01 to your computer and use it in GitHub Desktop.
Save ZhangzihanGit/3c04ffc30b141960f5162e7ffd348d01 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
select_words.py
# k * vocab
X_per_cluster = self.vectorizer_model.transform(concatenated_documents)
# D * vocab
X_origin = self.vectorizer_model.transform(origin_documents)
if self.word_select_method == 'tfidf_idfi':
socres = TFIDF_IDFi(X_per_cluster, X_origin, all_documents).socre()
elif self.word_select_method == 'tfidf_tfi':
socres = TFIDF_TFi(X_per_cluster, X_origin, all_documents).socre()
elif self.word_select_method == 'tfi':
socres = TFi(X_per_cluster).socre()
elif self.word_select_method == 'tfidfi':
socres = TFIDFi(X_per_cluster).socre()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment