[fix] kkma.post()에 빈 문자열 들어가지 않게 수정

pre-Q · Oct 10, 2023 · dbd3008 · dbd3008
1 parent fdbb1ef
commit dbd3008
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/extract_keyword.py b/extract_keyword.py
@@ -25,12 +25,13 @@ def get_keyword(application):
 
     output = list()
     for sample in data:
-        ex_pos = kkma.pos(sample)
-        text_data = []
-        for (text, tclass) in ex_pos:
-            if tclass == 'NNG' or tclass == 'NNP' or tclass == 'VV' or tclass == 'VA' or tclass == 'OL' or tclass == 'MAG':
-                text_data.append(text)
-        output.append(text_data)
+        if sample.strip():
+            ex_pos = kkma.pos(sample)
+            text_data = []
+            for (text, tclass) in ex_pos:
+                if tclass == 'NNG' or tclass == 'NNP' or tclass == 'VV' or tclass == 'VA' or tclass == 'OL' or tclass == 'MAG':
+                    text_data.append(text)
+            output.append(text_data)
 
     corpus = list(map(lambda x: ' '.join(x), output))