hy commited on
Commit
1ae484c
Β·
1 Parent(s): 4ff995f
Files changed (3) hide show
  1. aggro_model.py +3 -3
  2. crossref_model.py +1 -1
  3. main.py +4 -4
aggro_model.py CHANGED
@@ -200,15 +200,15 @@ def get_aggro_score(title: str) -> dict:
200
 
201
  # 5. λ“±κΈ‰ νŒμ •
202
  if final_score >= 80:
203
- reason = f"맀우 λ†’μŒ πŸ”΄ (AI: {bert_score:.0f}점, κ·œμΉ™: {rule_pattern})"
204
  recommendation = "μ „λ©΄ μˆ˜μ • ꢌμž₯"
205
 
206
  elif final_score >= 60:
207
- reason = f"λ†’μŒ 🟠 (AI: {bert_score:.0f}점, κ·œμΉ™: {rule_pattern})"
208
  recommendation = "κ³Όμž₯된 ν‘œν˜„ μˆ˜μ • ν•„μš”"
209
 
210
  elif final_score >= 40:
211
- reason = f"보톡 🟑 (AI: {bert_score:.0f}점, κ·œμΉ™: {rule_pattern})"
212
  recommendation = "일뢀 ν‘œν˜„ 쀑립화 ꢌμž₯"
213
 
214
  else:
 
200
 
201
  # 5. λ“±κΈ‰ νŒμ •
202
  if final_score >= 80:
203
+ reason = f"맀우 λ†’μŒ πŸ”΄ (AI: {bert_score:.2f}점, κ·œμΉ™: {rule_score:.0f}점,{rule_pattern})"
204
  recommendation = "μ „λ©΄ μˆ˜μ • ꢌμž₯"
205
 
206
  elif final_score >= 60:
207
+ reason = f"λ†’μŒ 🟠 (AI: {bert_score:.2f}점, κ·œμΉ™: {rule_score:.0f}점,{rule_pattern})"
208
  recommendation = "κ³Όμž₯된 ν‘œν˜„ μˆ˜μ • ν•„μš”"
209
 
210
  elif final_score >= 40:
211
+ reason = f"보톡 🟑 (AI: {bert_score:.2f}점, κ·œμΉ™: {rule_score:.0f}점,{rule_pattern})"
212
  recommendation = "일뢀 ν‘œν˜„ 쀑립화 ꢌμž₯"
213
 
214
  else:
crossref_model.py CHANGED
@@ -157,7 +157,7 @@ def get_crossref_score_and_reason(article_body: str) -> dict:
157
  recommendation = "κΈ°μ‚¬μ˜ 핡심 사싀관계가 타 μ–Έλ‘ μ‚¬μ—μ„œλ„ λ‹€λ£¨μ–΄μ§€λŠ”μ§€ 확인이 ν•„μš”ν•©λ‹ˆλ‹€."
158
 
159
  return {
160
- "score": max(0, min(1, final_score)),
161
  "reason": reason,
162
  "recommendation": recommendation,
163
  "paired_results": paired_results
 
157
  recommendation = "κΈ°μ‚¬μ˜ 핡심 사싀관계가 타 μ–Έλ‘ μ‚¬μ—μ„œλ„ λ‹€λ£¨μ–΄μ§€λŠ”μ§€ 확인이 ν•„μš”ν•©λ‹ˆλ‹€."
158
 
159
  return {
160
+ "score": max(0, min(1, round(final_score,4))),
161
  "reason": reason,
162
  "recommendation": recommendation,
163
  "paired_results": paired_results
main.py CHANGED
@@ -98,9 +98,9 @@ def analyze_article(request: ArticleRequest,db: Session = Depends(get_db)):
98
  )
99
 
100
  # μ΅œμ’… μœ„ν—˜λ„ 계산
101
- w_aggro = 0.4
102
- w_mismatch = 0.4
103
- w_crossref = 0.2
104
 
105
  final_score = (real_aggro.score * w_aggro) + \
106
  (real_mismatch.score * w_mismatch) + \
@@ -142,7 +142,7 @@ def analyze_article(request: ArticleRequest,db: Session = Depends(get_db)):
142
  # ------------------------------------------------
143
  # API λͺ…μ„Έμ„œ(AnalysisResponse) ν˜•μ‹μ— λ§žμΆ°μ„œ λ°˜ν™˜
144
  return AnalysisResponse(
145
- final_risk_score=round(final_score, 2), # μ†Œμˆ˜μ  2자리둜 반올림
146
  final_risk_level=final_level,
147
  breakdown={
148
  "aggro_score": real_aggro,
 
98
  )
99
 
100
  # μ΅œμ’… μœ„ν—˜λ„ 계산
101
+ w_aggro = 0.2
102
+ w_mismatch = 0.3
103
+ w_crossref = 0.5
104
 
105
  final_score = (real_aggro.score * w_aggro) + \
106
  (real_mismatch.score * w_mismatch) + \
 
142
  # ------------------------------------------------
143
  # API λͺ…μ„Έμ„œ(AnalysisResponse) ν˜•μ‹μ— λ§žμΆ°μ„œ λ°˜ν™˜
144
  return AnalysisResponse(
145
+ final_risk_score=round(final_score, 4), # μ†Œμˆ˜μ  4자리둜 반올림
146
  final_risk_level=final_level,
147
  breakdown={
148
  "aggro_score": real_aggro,