カテゴリ追加とスコア条件

kaminuma · Dec 10, 2024 · e025838 · e025838
1 parent d84c536
commit e025838
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/__init__.py b/app/__init__.py
@@ -32,7 +32,7 @@ def create_app():
     # カテゴリの定義
     app.config['CATEGORIES'] = [
     "運動", "仕事", "学習", "趣味", 
-    "食事", "睡眠", "買い物", "娯楽"
+    "食事", "睡眠", "買い物", "娯楽", "休憩", "家事"
     ]
 
     return app
diff --git a/app/analysis.py b/app/analysis.py
@@ -84,7 +84,7 @@ def fetch_data_by_period(start_date, end_date, user_id):
 
         # 各活動にカテゴリを追加
         data['category'] = [
-            result['labels'][0] ## if result['scores'][0] >= 0.5 else 'その他'
+            result['labels'][0] if result['scores'][0] >= 0.1 else 'その他'
             for result in classifications
         ]