Skip to content

Commit

Permalink
Merge pull request #253 from jiucchu/main
Browse files Browse the repository at this point in the history
[AI/FEAT] 클러스터링 결과를 memberId로 출력
  • Loading branch information
jiucchu authored Feb 14, 2024
2 parents e9e26be + ee7216f commit 601418c
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion AI/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ async def start_sbert():
return {'message' : "sbert is starting"}

class Item(BaseModel) :
name: str
memberId: str
sentence : str

class Item_and_num(BaseModel) :
Expand Down Expand Up @@ -79,4 +79,12 @@ async def start_sbert(data:Item_and_num):
while [] in copy_list:
copy_list.remove([])


for i in range(len(copy_list)):
for j in range (len(copy_list[i])):
target_value = copy_list[i][j]
for Item in input :
if Item.sentence == target_value :
copy_list[i][j] = Item.memberId

return copy_list

0 comments on commit 601418c

Please sign in to comment.