Spaces:
Sleeping
Sleeping
Kota Takahashi
commited on
Commit
•
a7b12c0
1
Parent(s):
b8973ef
コメント文を変更
Browse files- app.py +3 -3
- 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("
|
64 |
|
65 |
-
#
|
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]
|