iT邦幫忙

2022 iThome 鐵人賽

DAY 15
0
自我挑戰組

AI :PyTorch系列 第 15

第十五天 構建模型

  • 分享至 

  • xImage
  •  

前幾天我有學習到word embedding的定義
今天就來手寫一個embedding的class
先新建一個py文件
來構建模型
創建類

https://ithelp.ithome.com.tw/upload/images/20220929/20152487cQMIbKYgoE.png

添加forword的函數

https://ithelp.ithome.com.tw/upload/images/20220929/20152487xLzsOKjyYM.png

最後添加損失函數看看訓練效果

https://ithelp.ithome.com.tw/upload/images/20220929/20152487kN84BTlVXo.png

但是這樣子的程式來測試文本情感分類是會報錯的
原因是我們之前準備數據時候講內容存入dataset時候不應該是list 應該換成longtensor
在dataloader函數裡面將他改為longtensor就好了


上一篇
第十四天 文本序列化
下一篇
第十六天 循環神經網絡
系列文
AI :PyTorch30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言