DACON
search 구독 안내
알림 0

    MY XP

    XP 안내

    LEVEL 1

    다음 레벨까지 150 XP

    0/150 XP

    오늘의 XP

    0 / 800

    전체 XP

    0

    적립 XP

    0

    사용 XP

    0
    dacon logo

    SNS 계정으로 로그인

    google 구글 로그인 facebook 페이스북 로그인
    아직 데이콘 계정이 없나요? 회원가입
    메일 인증 필요

    로그인 하시려면 아래 이메일로 인증이 필요합니다. 이메일을 다시 보내시겠습니까?
    [데이콘] 회원가입 인증메일

    데이콘 회원가입을 환영합니다. 메일 인증은 데이콘 회원가입을 위한 필수 절차입니다. 아래 이메일을 인증하여 회원가입 절차를 완료하여 주시기 바랍니다.

    이용약관

    개인정보 처리방침

    마케팅 정보 수신 동의

    월간 데이콘 코드 유사성 판단 AI 경진대회

    알고리즘 | NLP | 분류 | 유사성 | Accuracy

    • moneyIcon 상금 : 총 600만 원
    • 908명 마감
    연습
    대회안내 데이터 코드 공유 토크 리더보드

     

    SBERT library를 활용한 baseline (LB : 0.839)

    둘리
    2022.05.27 18:05 2,420 조회 language
    comments 2

    https://www.sbert.net/를 참고해 작성한 baseline 코드입니다.
    데이터는 sample_train.csv만 사용해서 학습했습니다.
    huggingface나 pytorch에 비해 살짝 더 추상화가 되어 있어, 코드를 이해하고 사용하기 쉬운 반면 다양한 커스터마이징은 힘들다는 점이 단점입니다.
    predict 부분에서 버그(?)가 있는 것 같은데, 해결법을 아시는 분들은 공유해주시면 감사하겠습니다.

    코드
    댓글 2개
    로그인이 필요합니다
    0 / 1000
    ___
    2022.05.30 23:32

    CrossEncoder 모델을 save했는데 다시 불러와서 사용하는 방법을 모르겠습니다. load는 어떤 함수로 할 수 있나요?

    둘리
    2022.05.30 23:51

    loaded_model = CrossEncoder(저장된 모델 경로, num_labels=1)
    로 하시면 됩니다.

    📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!
    to_list 목록으로
    이전 글
    이번 대회에 대한 팁 몇 가지
    대회 - 월간 데이콘 코드 유사성 판단 AI 경진대회
    좋아요 38
    조회 3,447
    댓글 3
    3년 전
    현재 글
    SBERT library를 활용한 baseline (LB : 0.839)
    대회 - 월간 데이콘 코드 유사성 판단 AI 경진대회
    좋아요 11
    조회 2,420
    댓글 2
    3년 전
    다음 글
    [Baseline] CountVectorizer + Cosine Similarity
    대회 - 월간 데이콘 코드 유사성 판단 AI 경진대회
    좋아요 15
    조회 5,993
    댓글 0
    3년 전