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

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

Contributors 🐻

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

_{Jooho Kim}

_{Migyeong Kang}

_{Minjeong Kim}

_{Yejin Lee}

_{Jaebeen Lee}

_{Guhong Min}

_{Jieun Park}

_{Hyerin Lee}

jhtobigs/AMIOK