Skip to content

Latest commit

 

History

History
136 lines (109 loc) · 5.82 KB

README.md

File metadata and controls

136 lines (109 loc) · 5.82 KB
Raw

AM I OK?

전문의 답변 기반 심리진단 AI 🧐

2021.01.16. 제 11회 투빅스 빅데이터 컨퍼런스 발표작
  • 정신 질환 고민에 대한 질문글을 남기면, 진단을 통해 답변을 생성해 주는 서비스입니다.
  • TextRank로 질문을 추출 요약하고, Seq2seq과 T5를 통해 답변을 생성합니다.
  • 네이버 지식iN 전문의 답변을 토대로 구성되었으며, 상업적으로 이용할 의도가 전혀 없음을 밝힙니다.

Usage

1. Environment

1.1. Create virtual environment

$ conda create -n virtualenv
$ conda activate virtualenv

1.2. git Clone

$ git clone https://github.com/cryingjin/AMIOK.git
$ cd AMIOK

2. Install Packages & Download Models

2.1. Install Mecab

2.1.1. Install mecab-ko
$ tar xvfz mecab-0.996-ko-0.9.2.tar.gz
$ cd mecab-0.996-ko-0.9.2
$ ./configure
$ make
$ make check
$ sudo make install
2.1.2. Install mecab-ko-dic
$ tar xvfz mecab-ko-dic-2.1.1-20180720.tar.gz
$ cd mecab-ko-dic-2.1.1-20180720
$ ./configure
$ make
$ sudo make install
2.1.3. Install mecab-python
$ git clone https://bitbucket.org/eunjeon/mecab-python-0.996.git
$ cd mecab-python-0.996
$ python setup.py build
$ su
# python setup.py install

2.2. Install requirements

$ pip install -r requirements.txt

2.3. Install py-hanspell

$ git clone https://github.com/ssut/py-hanspell.git
$ cd py-hanspell 
$ python setup.py install

2.4. Download Data

drive의 tr_question_final.pickle, tr_answer_final.pickle 파일을 AMIOK/data/ 경로에 다운로드 받아주세요.

2.5. Get Pre-trained Checkpoints

drive의 seq2seq_ans3cut_epoch_10.pt, t5.h5 파일을 AMIOK/model/ 경로에 다운로드 받아주세요.

3. Run! ✨

$ python main.py -s "질문글을 입력해주세요 :>"

Results

Presentation

저희 프로젝트에 대해 자세하게 알고 싶으시다면, 하단의 링크를 참고해주세요!

  • GoogleDrive Badge
  • Youtube Badge

Contributors 🐻

빅데이터 동아리 ToBig's 멤버들이 함께한 프로젝트입니다.


Jooho Kim
Migyeong Kang
Minjeong Kim
Yejin Lee

Jaebeen Lee
Guhong Min
Jieun Park
Hyerin Lee