Pytorch LSTM实例2
生活随笔
收集整理的這篇文章主要介紹了
Pytorch LSTM实例2
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
對Pytorch中LSTM實例稍作修改,這是一個詞性標注的實例
#導入相應的包 import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optimtorch.manual_seed(1)#準備數據的階段 def prepare_sequence(seq, to_ix):idxs = [to_ix[w] for w in seq]return torch.tensor(idxs, dtype=torch.long)#DET 限定詞, NN 名詞 V 動詞 training_data = [("The dog ate the apple".split(), ["DET", "NN", "V", "DET", "NN"]),("Everybody read that book".split(), ["NN", "V", "DET", "NN"]) ] word_to_ix = {}for sent, tags in training_data:for word in sent:if word not in word_to_ix:word_to_ix[word] = len(word_to_ix) print(word_to_ix) #{'The': 0, 'dog': 1, 'ate': 2, 'the': 3, 'apple': 4, 'Everybody': 5, 'read': 6, 'that': 7, 'book': 8} tag_to_ix = {"總結
以上是生活随笔為你收集整理的Pytorch LSTM实例2的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Pytorch LSTM初识(详解L
- 下一篇: Pytorch 词嵌入word_emb