หน้าแรก>การเขียนโปรแกรมที่เกี่ยวข้อง>หลาม
ดาวน์โหลด

การแนะนำ

  • งานการจัดหมวดหมู่ใช้ใน Pytorch มีเครือข่ายประสาทบางส่วนในโมเดล
  • ฉันได้เปิดตัวรหัสแล้ว
    • รุ่นเก่า -17 เปิดตัวที่นี่
    • รุ่น Pytorch == 0.3.1 รุ่น ที่นี่

  • นี่คือสถาปัตยกรรมของฉันเอง --- pytorch-text-classification

  • bert สำหรับการจำแนกข้อความ --- pytorch_bert_text_classification

ความต้องการ 

 pyorch : 1.0.1
python : 3.6
torchtext: 0.2.1
cuda : 8.0 (support cuda speed up, can chose, default True)

การใช้งาน

แก้ไขไฟล์ config ดูไดเรกทอรี config (ที่นี่) สำหรับรายละเอียด 

 1、python main.py
2、python main.py --config_file ./Config/config.cfg 
3、sh run.sh

แบบอย่าง

มีเครือข่ายประสาทบางส่วนที่ใช้ใน Pytorch ดูแบบจำลองสำหรับรายละเอียด

ข้อมูล

SST-1 และ SST-2

ผลลัพธ์

ฉันไม่ได้ปรับพารามิเตอร์ไฮเปอร์อย่างจริงจังคุณสามารถดูเข้าสู่ระบบรถไฟได้ที่นี่

ความแม่นยำในชุดทดสอบต่อไปนี้ขึ้นอยู่กับความแม่นยำของชุด Dev ที่ดีที่สุด

ข้อมูล/รุ่น % SST-1 % SST-2
ซีเอ็นเอ็น 46.1086 84.2943
BI-LSTM 47.9186 86.3262
bi-gru 47.6923 86.7655

อ้างอิง

  • 基于 pytorch 的 cnn-lstm 神经网络模型调参小结
  • เครือข่ายประสาท Convolutional สำหรับการจำแนกประโยค
  • คุณสมบัติพจนานุกรมที่ไวต่อบริบทสำหรับการวิเคราะห์ความเชื่อมั่นของระบบประสาท
ขยาย
ข้อมูลเพิ่มเติม
แอปที่เกี่ยวข้อง
แนะนำสำหรับคุณ
ข้อมูลที่เกี่ยวข้อง ทั้งหมด