textcnn文本分类模型代码

以下是一个简单的 TextCNN 文本分类模型的代码示例：

import tensorflow as tf
from tensorflow.keras.layers import Input, Embedding, Conv1D, GlobalMaxPooling1D, Dense
from tensorflow.keras.models import Model

class TextCNN:
    def __init__(self, max_seq_len, vocab_size, embedding_dim, num_filters, filter_sizes, num_classes):
        self.max_seq_len = max_seq_len
        self.vocab_size = vocab_size
        self.embedding_dim = embedding_dim
        self.num_filters = num_filters
        self.filter_sizes = filter_sizes
        self.num_classes = num_classes

    def build_model(self):
        inputs = Input(shape=(self.max_seq_len,))
        embedding = Embedding(input_dim=self.vocab_size, output_dim=self.embedding_dim, input_length=self.max_seq_len)(inputs)
        conv_outputs = []
        for filter_size in self.filter_sizes:
            conv = Conv1D(filters=self.num_filters, kernel_size=filter_size, activation='relu')(embedding)
            pool = GlobalMaxPooling1D()(conv)
            conv_outputs.append(pool)
        concat = tf.keras.layers.concatenate(conv_outputs, axis=1)
        outputs = Dense(units=self.num_classes, activation='softmax')(concat)
        model = Model(inputs=inputs, outputs=outputs)
        return model

在上面的代码中，我们定义了一个名为 `TextCNN` 的类，它接收参数 `max_seq_len`、`vocab_size`、`embedding_dim`、`num_filters`、`filter_sizes` 和 `num_classes`。这些参数分别指定了输入序列的最大长度、词汇表大小、词向量维度、卷积核数量、卷积核大小列表和类别数。

我们在 `build_model` 方法中建立了模型。首先，我们使用 Keras 的 `Input` 层定义了输入张量。接着，我们使用 `Embedding` 层将每个单词映射为一个词向量，并且使用 `Conv1D` 层对词向量进行卷积操作。我们使用多个不同大小的卷积核来捕捉不同长度的 n-gram 特征。然后，我们使用 `GlobalMaxPooling1D` 层对每个卷积层的输出进行最大池化操作。最后，我们将所有卷积层的输出连接起来，并通过一个全连接层输出分类结果。

使用 TextCNN 模型进行文本分类时，需要将文本序列转换为数字序列，并将它们填充到相同的长度。可以使用 Keras 的 `Tokenizer` 类将文本序列转换为数字序列，并使用 Keras 的 `pad_sequences` 函数将它们填充到相同的长度。然后，使用 `model.fit` 函数训练模型即可。

注：以上代码仅为示例，实际应用中可能需要根据具体情况进行调整。