2022 iThome 鐵人賽
分享至
前幾天我有學習到word embedding的定義今天就來手寫一個embedding的class先新建一個py文件來構建模型創建類
添加forword的函數
最後添加損失函數看看訓練效果
但是這樣子的程式來測試文本情感分類是會報錯的原因是我們之前準備數據時候講內容存入dataset時候不應該是list 應該換成longtensor在dataloader函數裡面將他改為longtensor就好了
IT邦幫忙