코드 소개:
위 코드는 transformer 모델을 사용한 챗봇에 칵테일 추천에 필요한 데이터 셋을 학습시켜 본팀이 제시한 시나리오대로 기분, 맛, 도수, 재료를 입력받으며 소통하는 챗봇 토이프로젝트의 코드이다.
colab으로 챗봇 실행하는 방법:
-
구글 드라이브에 폴더를 생성하고 ChatbotData.csv와 bar.csv를 넣는다.
-
다음 경로명을 코드의 빈칸에 넣어준다.
/content/drive/MyDrive/생성한 폴더이름/ChatbotData.csv
/content/drive/MyDrive/생성한 폴더이름/bar.csv
#기존 챗봇 데이터셋 불러오기 data = pd.read_csv('____________')
#칵테일 추천을 위한 데이터셋 불러오기 bar_data=pd.read_csv('____________')
예시
- 런타임-런타임 유형 변경으로 들어가 하드웨어 가속기를 GPU로 설정해준다.
- 실행 버튼을 눌러 실행해준다.