python code

traP-jp · Aug 18, 2024 · 36f9606 · 36f9606
1 parent 54afd61
commit 36f9606
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/tag/keytest.py b/src/tag/keytest.py
@@ -0,0 +1 @@
+import pke
diff --git a/src/tag/keyword.py b/src/tag/keyword.py
@@ -0,0 +1,9 @@
+import pke
+
+extractor = pke.unsupervised.TopicRank()
+
+def keyword_extract(text, n):
+    extractor.load_document(input=text, language='ja')
+    extractor.candidate_selection(pos={'NOUN', 'PROPN'})
+    extractor.candidate_weighting()
+    return extractor.get_n_best(n=n)