Kota Takahashi commited on
Commit
a7b12c0
1 Parent(s): b8973ef

コメント文を変更

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. cosine_similarity_calculator.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from tfidf_calculator import JapaneseTextVectorizer
5
  from cosine_similarity_calculator import CosineSimilarityCalculator
6
  from summerizer import TextSummarizer
7
 
8
- st.title("ニュース検索アプリ")
9
 
10
  # 初期化
11
  best_article_text = None
@@ -60,9 +60,9 @@ if st.session_state['news_fetched']:
60
  result_word_similarity = word_similarity.calculate_similarity(search_word, article_keyword_list)
61
 
62
  if result_word_similarity is None:
63
- raise ValueError("類似度計算結果がNoneです。名詞を変更して再度試してください。")
64
 
65
- # None でない値のみを抽出
66
  filtered_data = {k: v for k, v in result_word_similarity.items() if v is not None}
67
 
68
  # 最大値を持つキーとその値を取得
 
5
  from cosine_similarity_calculator import CosineSimilarityCalculator
6
  from summerizer import TextSummarizer
7
 
8
+ st.title("最新ニュース要約アプリ")
9
 
10
  # 初期化
11
  best_article_text = None
 
60
  result_word_similarity = word_similarity.calculate_similarity(search_word, article_keyword_list)
61
 
62
  if result_word_similarity is None:
63
+ raise ValueError("単語の類似度を計算できませんでした。名詞を変更して再度試してください。")
64
 
65
+ # cos類似度の計算結果
66
  filtered_data = {k: v for k, v in result_word_similarity.items() if v is not None}
67
 
68
  # 最大値を持つキーとその値を取得
cosine_similarity_calculator.py CHANGED
@@ -52,7 +52,7 @@ class CosineSimilarityCalculator:
52
  - search_word (str): 検索ワード
53
  - article_keyword_list (list): 記事のキーワードリスト
54
  Returns:
55
- - similarities (dict): 記事キーワードとそれぞれの検索ワードのコサイン類似度を含む辞書を作成。
56
  モデルにない単語の場合はNoneを返す
57
  """
58
  # 検索ワードの埋め込みベクトルを取得
@@ -62,7 +62,7 @@ class CosineSimilarityCalculator:
62
  return None
63
 
64
  similarities = {}
65
- # 記事キーワードの埋め込みベクトルを取得し、コサイン類似度を計算
66
  for keyword in article_keyword_list:
67
  if keyword in self.model.wv:
68
  keyword_embedding = self.model.wv[keyword]
 
52
  - search_word (str): 検索ワード
53
  - article_keyword_list (list): 記事のキーワードリスト
54
  Returns:
55
+ - similarities (dict): 記事キーワードとそれぞれの検索ワードのコサイン類似度を含むdictを作成。
56
  モデルにない単語の場合はNoneを返す
57
  """
58
  # 検索ワードの埋め込みベクトルを取得
 
62
  return None
63
 
64
  similarities = {}
65
+ # 記事キーワードの埋め込みベクトルを取得し、cos類似度を計算
66
  for keyword in article_keyword_list:
67
  if keyword in self.model.wv:
68
  keyword_embedding = self.model.wv[keyword]