使用pytorch实现CNN和LSTM并列的文本分类代码并训练数据

最新推荐文章于 2023-10-24 16:08:49 发布

weixin_42601702

最新推荐文章于 2023-10-24 16:08:49 发布

阅读量274

点赞数

文章标签： pytorch cnn lstm 分类深度学习

本文链接：https://blog.csdn.net/weixin_42601702/article/details/129509387

版权

使用 PyTorch 实现 CNN 和 LSTM 并列的文本分类的代码可以这样写：

import torch
import torch.nn as nn

class TextClassifier(nn.Module):
    def __init__(self, vocab_size, embedding_dim, num_class):
        super().__init__()
        self.embedding = nn.Embedding(vocab_size, embedding_dim)
        self.cnn = nn.Conv1d(e