We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4b90c20 commit 8733539Copy full SHA for 8733539
tests/test_text_mining.py
@@ -2,13 +2,16 @@
2
from unittest import TestCase
3
import numpy as np
4
import pandas as pd
5
+import nltk
6
7
from niaarm.niaarm import _cut_point
8
from niaarm.text import Corpus, TextRule, NiaARTM
9
10
11
class TestTextMining(TestCase):
12
def setUp(self):
13
+ nltk.download('punkt')
14
+ nltk.download('stopwords')
15
ds_path = os.path.join(os.path.dirname(__file__), 'test_data', 'artm_test_dataset.json')
16
df = pd.read_json(ds_path, orient='records')
17
documents = df['text'].tolist()
0 commit comments