tf_seq2seq_chatbot
1.0.0
หมายเหตุ: พื้นที่เก็บข้อมูลไม่ได้รับการดูแลรักษา อย่าลังเลที่จะ PM ฉันหากคุณต้องการเข้ารับการบำรุงรักษา
สร้างแชทบอตการสนทนาทั่วไปโดยยึดตามแนวทาง hot seq2seq ที่นำมาใช้ในเทนเซอร์โฟลว์ เนื่องจากจนถึงขณะนี้ยังไม่ได้ผลลัพธ์ที่ดี โปรดพิจารณาการใช้งาน seq2seq อื่นๆ ด้วย
ผลลัพธ์ปัจจุบันค่อนข้างแย่:
hello baby - hello
how old are you ? - twenty .
i am lonely - i am not
nice - you ' re not going to be okay .
so rude - i ' m sorry .
ข้อสงวนสิทธิ์:
เรายินดีต้อนรับทุกคนให้ตรวจสอบโค้ดและเสนอแนะการปรับปรุง
การกระทำที่เกิดขึ้นจริง
เอกสาร
ภาพสวย
ความอนุเคราะห์ของบทความนี้
ตั้งค่า
git clone [email protected]:nicolas-ivanov/tf_seq2seq_chatbot.git
cd tf_seq2seq_chatbot
bash setup.sh
วิ่ง
ฝึกฝนโมเดล seq2seq บนคลังคำบรรยายภาพยนตร์ขนาดเล็ก (17 MB):
python train.py
(คำสั่งนี้จะรันการฝึกบน CPU... คำสั่ง GPU กำลังมา)
ทดสอบโมเดลที่ได้รับการฝึกอบรมกับชุดคำถามทั่วไป:
python test.py
แชทกับโมเดลที่ผ่านการฝึกอบรมในคอนโซล:
python chat.py
พารามิเตอร์การกำหนดค่าทั้งหมดจะถูกเก็บไว้ที่ tf_seq2seq_chatbot/configs/config.py
การใช้งาน GPU
หากคุณโชคดีที่มีการกำหนดค่า GPU ที่เหมาะสมสำหรับเทนเซอร์โฟลว์อยู่แล้ว สิ่งนี้น่าจะได้ผล:
python train.py
มิฉะนั้นคุณอาจต้องสร้าง tensorflow จากแหล่งที่มาและรันโค้ดดังนี้:
cd tensorflow # cd to the tensorflow source folder
cp -r ~/tf_seq2seq_chatbot ./ # copy project's code to tensorflow root
bazel build -c opt --config=cuda tf_seq2seq_chatbot:train # build with gpu-enable option
./bazel-bin/tf_seq2seq_chatbot/train # run the built code
ความต้องการ
