模型代码如下所示
import jieba
import torch
import torch.nn as nn
import torch.nn.functional as functions
import numpy as np
import time
# -*- coding: utf-8 -*-
#
# coding: utf-8
class TextCNN_Model(nn.Module):
def __init__(self, configs):
super(TextCNN_Model, self).__init__()
vocab_size = configs.vocab_size
embedding_dimension = configs.embedding_dimension
label_num = configs.label_num
self.sentence_len=210
self.p_embed_n=25
self.out_num=50
self.relation_dic = np.load("rela_dic.npy", allow_pickle=True).item()
self.conv0 = nn.Sequential(
nn.Conv1d(in_channels=2*self.p_embed_n+embedding_dimension,out_channels=self.out_num,kernel_size=(3),bias=False,padding=(1)),
nn.LayerNorm([self.sentence_len]),
nn.ReLU(),
)
self.conv1 = nn.Sequential(
nn.Conv1d(in_channels&#