vumichien commited on
Commit
607ef0e
·
1 Parent(s): 668c994

add similarity score

Browse files
Files changed (2) hide show
  1. models.py +1 -0
  2. routes/predict.py +1 -0
models.py CHANGED
@@ -40,6 +40,7 @@ class PredictResult(BaseModel):
40
  standard_subject: str
41
  standard_name: str
42
  anchor_name: str
 
43
 
44
 
45
  class PredictRawRequest(BaseModel):
 
40
  standard_subject: str
41
  standard_name: str
42
  anchor_name: str
43
+ similarity_score: float
44
 
45
 
46
  class PredictRawRequest(BaseModel):
routes/predict.py CHANGED
@@ -183,6 +183,7 @@ async def predict_raw(
183
  standard_subject=row["標準科目"],
184
  standard_name=row["標準項目名"],
185
  anchor_name=row["基準名称"],
 
186
  )
187
  results.append(result)
188
 
 
183
  standard_subject=row["標準科目"],
184
  standard_name=row["標準項目名"],
185
  anchor_name=row["基準名称"],
186
+ similarity_score=float(row["基準名称類似度"]),
187
  )
188
  results.append(result)
189