Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- inference.py +0 -1
inference.py
CHANGED
|
@@ -212,7 +212,6 @@ def classify_ma(sentence: str) -> tuple[int, torch.Tensor, torch.Tensor]:
|
|
| 212 |
def infer_topic(new_text: str) -> tuple[np.ndarray, float]:
|
| 213 |
model_trained = tp.CTModel.load(str(topic_model_trained))
|
| 214 |
new_word_list = tokenizer_sudachi.get_wakati(new_text)
|
| 215 |
-
print(new_word_list) #この一文を追加
|
| 216 |
new_doc = model_trained.make_doc(new_word_list)
|
| 217 |
topic_dist, ll = model_trained.infer(new_doc)
|
| 218 |
return topic_dist,new_word_list, ll
|
|
|
|
| 212 |
def infer_topic(new_text: str) -> tuple[np.ndarray, float]:
|
| 213 |
model_trained = tp.CTModel.load(str(topic_model_trained))
|
| 214 |
new_word_list = tokenizer_sudachi.get_wakati(new_text)
|
|
|
|
| 215 |
new_doc = model_trained.make_doc(new_word_list)
|
| 216 |
topic_dist, ll = model_trained.infer(new_doc)
|
| 217 |
return topic_dist,new_word_list, ll
|