데이터의 Q 필드를 <usr>발화, A 필드를 <sys>발화 그리고 감정 레이블을 <sent>로 매핑해 P(<sys>|<usr>, <sent>)를 최대화 할 수 있는 모델을 학습한다. 감정 레이블은 이곳의 정의를 따른다(일상다반사 0, 이별(부정) 1, 사랑(긍정) 2).
CUDA_VISIBLE_DEVICES=0 python train.py --chat
user > 방가워용.
Simsimi > 저도요.
user > 어버이날엔 뭘 하죠?
Simsimi > 부모님께 효도하는 것도 좋을 것 같아요.
user > 용돈이 너무 적어요.
Simsimi > 아껴 쓰세요.
user > 미친듯이 고민해도 방법이 없네요.
Simsimi > 좋은 방법이 있을 거예요.
user > 큼지막한 방에서
Simsimi > 혼자만의 시간을 가지세요.
user > 큰돈을 만드는 방법은?
Simsimi > 돈을 많이 버는 것뿐이에요.
user > 집은 너무 비싼거 같아요.
Simsimi > 비싸다고 해서 다 좋은 건 아니에요.