ogawaal commited on
Commit
246a58a
·
verified ·
1 Parent(s): 49ebd54

Update inference.py

Browse files
Files changed (1) hide show
  1. 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