/AMIOK

[제 11회 투빅스 컨퍼런스] AM I OK ? - 전문의 답변 기반 심리진단 AI

Primary LanguagePythonMIT LicenseMIT

AM I OK?

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

2021.01.16. 제 11회 투빅스 빅데이터 컨퍼런스 발표작

  • 정신 질환 고민에 대한 질문글을 남기면, 진단을 통해 답변을 생성해 주는 서비스입니다.
  • TextRank로 질문을 추출 요약하고, Seq2seqT5를 통해 답변을 생성합니다.
  • 네이버 지식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

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

2.5. Get Pre-trained Checkpoints

driveseq2seq_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