Chanjeans commited on
Commit
59c2c82
·
verified ·
1 Parent(s): b33b651

현진_임계값 수정

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1185,9 +1185,9 @@ def predict_depression(text: str):
1185
  kobert_score = max(0.35, min(kobert_score, 0.88))
1186
  xgb_proba = max(0.3, min(xgb_proba, 0.83))
1187
  combined_score = (kobert_score * 0.55) + (xgb_proba * 0.45)
1188
- if combined_score > 0.75:
1189
  label = "상담 권장"
1190
- elif combined_score > 0.7:
1191
  label = "관심 필요"
1192
  else:
1193
  label = "정상"
 
1185
  kobert_score = max(0.35, min(kobert_score, 0.88))
1186
  xgb_proba = max(0.3, min(xgb_proba, 0.83))
1187
  combined_score = (kobert_score * 0.55) + (xgb_proba * 0.45)
1188
+ if combined_score > 0.72:
1189
  label = "상담 권장"
1190
+ elif combined_score > 0.68:
1191
  label = "관심 필요"
1192
  else:
1193
  label = "정상"