Spaces:
Runtime error
Runtime error
Update Space
Browse files
app.py
CHANGED
@@ -43,11 +43,13 @@ model.load_state_dict(model_state)
|
|
43 |
model.eval()
|
44 |
|
45 |
# Fungsi untuk tokenisasi, stemming, dan bag-of-words
|
|
|
|
|
46 |
def tokenize(sentence):
|
47 |
return nltk.word_tokenize(sentence)
|
48 |
|
49 |
def stem(word):
|
50 |
-
return
|
51 |
|
52 |
def bag_of_words(tokenized_sentence, all_words):
|
53 |
tokenized_sentence = [stem(w) for w in tokenized_sentence]
|
@@ -55,6 +57,7 @@ def bag_of_words(tokenized_sentence, all_words):
|
|
55 |
for idx, w in enumerate(all_words):
|
56 |
if w in tokenized_sentence:
|
57 |
bag[idx] = 1.0
|
|
|
58 |
return bag
|
59 |
|
60 |
# Fungsi prediksi
|
|
|
43 |
model.eval()
|
44 |
|
45 |
# Fungsi untuk tokenisasi, stemming, dan bag-of-words
|
46 |
+
stemmer = PorterStemmer()
|
47 |
+
|
48 |
def tokenize(sentence):
|
49 |
return nltk.word_tokenize(sentence)
|
50 |
|
51 |
def stem(word):
|
52 |
+
return stemmer.stem(word.lower())
|
53 |
|
54 |
def bag_of_words(tokenized_sentence, all_words):
|
55 |
tokenized_sentence = [stem(w) for w in tokenized_sentence]
|
|
|
57 |
for idx, w in enumerate(all_words):
|
58 |
if w in tokenized_sentence:
|
59 |
bag[idx] = 1.0
|
60 |
+
|
61 |
return bag
|
62 |
|
63 |
# Fungsi prediksi
|