Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat - 페르소나 추출 모델을 위한 베이스라인 추가 #5

Merged
merged 3 commits into from
Mar 29, 2024
Merged

feat - 페르소나 추출 모델을 위한 베이스라인 추가 #5

merged 3 commits into from
Mar 29, 2024

Conversation

DonghaeSuh
Copy link
Contributor

Overview

  • 페르소나 추출 모델을 위한 pytorch-lightening 기반 베이스라인 추가

Change Log

  • 모델 학습

    • ml/persona_extraction/kobart_train.py
    • ml/persona_extraction/et5_train.py

  • 모델 config

    • ml/persona_extraction/configs/et5/et5-banmal-by-bleu-config.json 외 5개

  • 버전 관리

    • ml/persona_extraction/requirements.yaml

  • 가이드

    • ml/persona_extraction/README.md

To Reviewer

  • et5 모델 같은 경우, 다음 ETRI에서 신청하여 받을 수 있음
  • 학습 데이터 셋의 경우, 다음 AI Hub 한국어 멀티세션 대화에서 받을 수 있음
  • 모델 학습을 위해 반말 변환한 데이터 셋의 경우 다음 Huggingface dataset에서 받을 수 있음
  • 학습은 readme를 따라 하면 가능 (테스트 완료)
  • 본 코드는 다음 개인 레포지토리 링크에서 작성되고 수정되었음. 세부 commit들은 해당 레포지토리에서 확인 가능
  • 학습에 대한 자세한 설명은 다음 블로그 링크에 정리해놓았음

Issue Tags

@DonghaeSuh DonghaeSuh self-assigned this Mar 14, 2024
@DonghaeSuh DonghaeSuh changed the title FEAT - 페르소나 추출 모델을 위한 베이스라인 추가 feat - 페르소나 추출 모델을 위한 베이스라인 추가 Mar 27, 2024
@iloveonsen
Copy link
Contributor

동해님 방식대로 ml 폴더 내부에 각자하신 내용을 정리하는것으로 하겠습니다~

@iloveonsen iloveonsen merged commit e2b12b6 into main Mar 29, 2024
@iloveonsen iloveonsen deleted the feat/4/baseline branch March 29, 2024 01:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@DonghaeSuh DonghaeSuh

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

[FEAT] 페르소나 추출 모델 학습 베이스라인
2 participants
@DonghaeSuh @iloveonsen