华为开源自研AI框架昇思MindSpore应用实践：RNN实现情感分类

╰半橙微兮° 2024-03-30 09:12 62阅读 0赞

#### 目录 ####

*  一、环境准备
 *   *  1.进入ModelArts官网
     *  2.使用CodeLab体验Notebook实例
 *  二、数据准备
 *   *  1.数据下载模块
     *  2.加载IMDB数据集
     *  2.加载预训练词向量
 *  三、数据集预处理
 *  四、模型构建
 *   *  1.Embedding
     *  2.RNN(循环神经网络)
     *  3.Dense
     *  4.损失函数与优化器
     *  5.训练逻辑
     *  6.评估指标和逻辑
 *  五、模型训练与保存
 *  六、模型加载与测试
 *  七、自定义输入测试

本教程是通过示例代码说明情感分类是自然语言处理中的经典任务，是典型的分类问题。使用MindSpore实现一个基于RNN网络的情感分类模型，实现如下的效果：

输入: This film is terrible
    正确标签: Negative
    预测标签: Negative
    
    输入: This film is great
    正确标签: Positive
    预测标签: Positive

如果你对MindSpore感兴趣，可以关注[**昇思MindSpore社区**][MindSpore]

![在这里插入图片描述][a9571b75523b4cbca146948dbadf8715.png]

## 一、环境准备 ##

### 1.进入ModelArts官网 ###

云平台帮助用户快速创建和部署模型，管理全周期AI工作流，选择下面的云平台以开始使用昇思MindSpore，可以在昇思教程中进入[**ModelArts官网**][ModelArts]

![在这里插入图片描述][ee2578b7b26f4435bfd12aedd8959967.png]

选择下方CodeLab立即体验

![在这里插入图片描述][547900ad31bf4f4386132554fe26dfb9.png]  
等待环境搭建完成  
![在这里插入图片描述][1466f73e44fa4fe69bd38b0b30661e22.png]

### 2.使用CodeLab体验Notebook实例 ###

下载[**NoteBook样例代码**][NoteBook]，`.ipynb`为样例代码

![在这里插入图片描述][19d112a9f67e4b48ae558cc0a4a8ce1e.png]

选择ModelArts Upload Files上传`.ipynb`文件

![在这里插入图片描述][e61bd133ba3542f09c446844d843b8fb.png]

![在这里插入图片描述][5979276a6ed045c580aa4f40b2209f4d.png]

![在这里插入图片描述][a59b36db7dac45079c8d194c1a8b504b.png]

选择Kernel环境

![在这里插入图片描述][550d87512bab4b3a9c4d13138c32769d.png]

进入[**昇思MindSpore官网**][MindSpore 1]，点击上方的安装

![在这里插入图片描述][2dec09846d164105b79d514aa0deed82.png]

获取安装命令

![在这里插入图片描述][744a4ea99b0b4cba82dbbfd9f290e402.png]

回到Notebook中，在第一块代码前加入三块命令  
![在这里插入图片描述][d1b5e7f5355b410599f32ed6a59d9000.png]

pip install --upgrade pip

conda install mindspore-gpu=1.9.0 cudatoolkit=10.1 -c mindspore -c conda-forge

pip install mindvision

依次运行即可

![在这里插入图片描述][7eec4a2d97634beeae03da93eabf8eb5.png]

![在这里插入图片描述][62986501180f4c44a9cbda230df95070.png]

## 二、数据准备 ##

本节使用情感分类的经典数据集[IMDB影评数据集][IMDB]，数据集包含Positive和Negative两类，下面为其样例：

<table> 
 <thead> 
  <tr> 
   <th>Review</th> 
   <th>Label</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>“Quitting” may be as much about exiting a pre-ordained identity as about drug withdrawal. As a rural guy coming to Beijing, class and success must have struck this young artist face on as an appeal to separate from his roots and far surpass his peasant parents’ acting success. Troubles arise, however, when the new man is too new, when it demands too big a departure from family, history, nature, and personal identity. The ensuing splits, and confusion between the imaginary and the real and the dissonance between the ordinary and the heroic are the stuff of a gut check on the one hand or a complete escape from self on the other.</td> 
   <td>Negative</td> 
  </tr> 
  <tr> 
   <td>This movie is amazing because the fact that the real people portray themselves and their real life experience and do such a good job it’s like they’re almost living the past over again. Jia Hongsheng plays himself an actor who quit everything except music and drugs struggling with depression and searching for the meaning of life while being angry at everyone especially the people who care for him most…</td> 
   <td>Positive</td> 
  </tr> 
 </tbody> 
</table>

此外，需要使用预训练词向量对自然语言单词进行编码，以获取文本的语义特征，本节选取[Glove][]词向量作为Embedding。

### 1.数据下载模块 ###

为了方便数据集和预训练词向量的下载，首先设计数据下载模块，实现可视化下载流程，并保存至指定路径。数据下载模块使用`requests`库进行http请求，并通过`tqdm`库对下载百分比进行可视化。此外针对下载安全性，使用IO的方式下载临时文件，而后保存至指定的路径并返回。

`tqdm`和`requests`库需手动安装，命令如下：`pip install tqdm requests`

![在这里插入图片描述][e5856d8803a540f1a60701cacdebc8fa.png]

import os
    import shutil
    import requests
    import tempfile
    from tqdm import tqdm
    from typing import IO
    from pathlib import Path
    
    # 指定保存路径为 `home_path/.mindspore_examples`
    cache_dir = Path.home() / '.mindspore_examples'
    
    def http_get(url: str, temp_file: IO):
        """使用requests库下载数据，并使用tqdm库进行流程可视化"""
        req = requests.get(url, stream=True)
        content_length = req.headers.get('Content-Length')
        total = int(content_length) if content_length is not None else None
        progress = tqdm(unit='B', total=total)
        for chunk in req.iter_content(chunk_size=1024):
            if chunk:
                progress.update(len(chunk))
                temp_file.write(chunk)
        progress.close()
    
    def download(file_name: str, url: str):
        """下载数据并存为指定名称"""
        if not os.path.exists(cache_dir):
            os.makedirs(cache_dir)
        cache_path = os.path.join(cache_dir, file_name)
        cache_exist = os.path.exists(cache_path)
        if not cache_exist:
            with tempfile.NamedTemporaryFile() as temp_file:
                http_get(url, temp_file)
                temp_file.flush()
                temp_file.seek(0)
                with open(cache_path, 'wb') as cache_file:
                    shutil.copyfileobj(temp_file, cache_file)
        return cache_path

![在这里插入图片描述][f76f5b6dd26e46d5bb7fec945173db5a.png]

完成数据下载模块后，下载IMDB数据集进行测试(此处使用华为云的镜像用于提升下载速度)。下载过程及保存的路径如下：

![在这里插入图片描述][81256546a2d34dc3a29bafb8273154b4.png]

### 2.加载IMDB数据集 ###

下载好的IMDB数据集为tar.gz文件，我们使用Python的tarfile库对其进行读取，并将所有数据和标签分别进行存放。原始的IMDB数据集解压目录如下：

├── aclImdb
    │   ├── imdbEr.txt
    │   ├── imdb.vocab
    │   ├── README
    │   ├── test
    │   └── train
    │         ├── neg
    │         ├── pos
    ...

数据集已分割为train和test两部分，且每部分包含neg和pos两个分类的文件夹，因此需分别train和test进行读取并处理数据和标签。

import re
    import six
    import string
    import tarfile
    
    class IMDBData():
        """IMDB数据集加载器
    
        加载IMDB数据集并处理为一个Python迭代对象。
    
        """
        label_map = {
        
            "pos": 1,
            "neg": 0
        }
        def __init__(self, path, mode="train"):
            self.mode = mode
            self.path = path
            self.docs, self.labels = [], []
    
            self._load("pos")
            self._load("neg")
    
        def _load(self, label):
            pattern = re.compile(r"aclImdb/{}/{}/.*\.txt$".format(self.mode, label))
            # 将数据加载至内存
            with tarfile.open(self.path) as tarf:
                tf = tarf.next()
                while tf is not None:
                    if bool(pattern.match(tf.name)):
                        # 对文本进行分词、去除标点和特殊字符、小写处理
                        self.docs.append(str(tarf.extractfile(tf).read().rstrip(six.b("\n\r"))
                                             .translate(None, six.b(string.punctuation)).lower()).split())
                        self.labels.append([self.label_map[label]])
                    tf = tarf.next()
    
        def __getitem__(self, idx):
            return self.docs[idx], self.labels[idx]
    
        def __len__(self):
            return len(self.docs)

![在这里插入图片描述][91116e702e9e4364871b442c0a4da1cf.png]  
完成IMDB数据加载器后，加载训练数据集进行测试，输出数据集数量：

![在这里插入图片描述][6bb202235b0347319c1e421ac9b95426.png]

将IMDB数据集加载至内存并构造为迭代对象后，可以使用`mindspore.dataset`提供的`Generatordataset`接口加载数据集迭代对象，并进行下一步的数据处理，下面封装一个函数将train和test分别使用`Generatordataset`进行加载，并指定数据集中文本和标签的`column_name`分别为`text`和`label`:

import mindspore.dataset as ds
    
    def load_imdb(imdb_path):
        imdb_train = ds.GeneratorDataset(IMDBData(imdb_path, "train"), column_names=["text", "label"], shuffle=True)
        imdb_test = ds.GeneratorDataset(IMDBData(imdb_path, "test"), column_names=["text", "label"], shuffle=False)
        return imdb_train, imdb_test

![在这里插入图片描述][14a0e215164545c3b3b80697ea1aae46.png]

加载IMDB数据集，可以看到`imdb_train`是一个GeneratorDataset对象。

imdb_train, imdb_test = load_imdb(imdb_path)
    imdb_train

![在这里插入图片描述][c2f5f0ed47b44991af0aeb071300bdd6.png]

### 2.加载预训练词向量 ###

预训练词向量是对输入单词的数值化表示，通过`nn.Embedding`层，采用查表的方式，输入单词对应词表中的index，获得对应的表达向量。 因此进行模型构造前，需要将Embedding层所需的词向量和词表进行构造。这里我们使用Glove(Global Vectors for Word Representation)这种经典的预训练词向量， 其数据格式如下：

<table> 
 <thead> 
  <tr> 
   <th>Word</th> 
   <th>Vector</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>the</td> 
   <td>0.418 0.24968 -0.41242 0.1217 0.34527 -0.044457 -0.49688 -0.17862 -0.00066023 …</td> 
  </tr> 
  <tr> 
   <td>,</td> 
   <td>0.013441 0.23682 -0.16899 0.40951 0.63812 0.47709 -0.42852 -0.55641 -0.364 …</td> 
  </tr> 
 </tbody> 
</table>

我们直接使用第一列的单词作为词表，使用`dataset.text.Vocab`将其按顺序加载；同时读取每一行的Vector并转为`numpy.array`，用于`nn.Embedding`加载权重使用。具体实现如下：

import zipfile
    import numpy as np
    
    def load_glove(glove_path):
        glove_100d_path = os.path.join(cache_dir, 'glove.6B.100d.txt')
        if not os.path.exists(glove_100d_path):
            glove_zip = zipfile.ZipFile(glove_path)
            glove_zip.extractall(cache_dir)
    
        embeddings = []
        tokens = []
        with open(glove_100d_path, encoding='utf-8') as gf:
            for glove in gf:
                word, embedding = glove.split(maxsplit=1)
                tokens.append(word)
                embeddings.append(np.fromstring(embedding, dtype=np.float32, sep=' '))
        # 添加 <unk>, <pad> 两个特殊占位符对应的embedding
        embeddings.append(np.random.rand(100))
        embeddings.append(np.zeros((100,), np.float32))
    
        vocab = ds.text.Vocab.from_list(tokens, special_tokens=["<unk>", "<pad>"], special_first=False)
        embeddings = np.array(embeddings).astype(np.float32)
        return vocab, embeddings

![在这里插入图片描述][c4814c22dd2346c1937e47a3b427a130.png]

由于数据集中可能存在词表没有覆盖的单词，因此需要加入`<unk>`标记符；同时由于输入长度的不一致，在打包为一个batch时需要将短的文本进行填充，因此需要加入`<pad>`标记符。完成后的词表长度为原词表长度+2。

下面下载Glove词向量，并加载生成词表和词向量权重矩阵。

glove_path = download('glove.6B.zip', 'https://mindspore-website.obs.myhuaweicloud.com/notebook/datasets/glove.6B.zip')
    vocab, embeddings = load_glove(glove_path)
    len(vocab.vocab())

![在这里插入图片描述][750538bed2b74bab80e12264d092ac69.png]

使用词表将`the`转换为index id，并查询词向量矩阵对应的词向量：

idx = vocab.tokens_to_ids('the')
    embedding = embeddings[idx]
    idx, embedding

![在这里插入图片描述][c815d283678945a1bf18affe273a6b02.png]

## 三、数据集预处理 ##

通过加载器加载的IMDB数据集进行了分词处理，但不满足构造训练数据的需要，因此要对其进行额外的预处理。其中包含的预处理如下：

*  通过Vocab将所有的Token处理为index id。
 *  将文本序列统一长度，不足的使用`<pad>`补齐，超出的进行截断。

这里我们使用`mindspore.dataset`中提供的接口进行预处理操作。这里使用到的接口均为MindSpore的高性能数据引擎设计，每个接口对应操作视作数据流水线的一部分，详情请参考[MindSpore数据引擎][MindSpore 2]。 首先针对token到index id的查表操作，使用`text.Lookup`接口，将前文构造的词表加载，并指定`unknown_token`。其次为文本序列统一长度操作，使用`PadEnd`接口，此接口定义最大长度和补齐值(`pad_value`)，这里我们取最大长度为500，填充值对应词表中`<pad>`的index id。

> 除了对数据集中`text`进行预处理外，由于后续模型训练的需要，要将`label`数据转为float32格式。

import mindspore as ms
    
    lookup_op = ds.text.Lookup(vocab, unknown_token='<unk>')
    pad_op = ds.transforms.PadEnd([500], pad_value=vocab.tokens_to_ids('<pad>'))
    type_cast_op = ds.transforms.TypeCast(ms.float32)

![在这里插入图片描述][68ed4083438e4e328227a25cd22f206d.png]

完成预处理操作后，需将其加入到数据集处理流水线中，使用map接口对指定的column添加操作。

![在这里插入图片描述][deb57adf87924a10be7dc309095682c2.png]

由于IMDB数据集本身不包含验证集，我们手动将其分割为训练和验证两部分，比例取0.7, 0.3。

imdb_train, imdb_valid = imdb_train.split([0.7, 0.3])

![在这里插入图片描述][c9af93b4358140caa6da5988109d0c6a.png]

最后指定数据集的`batch`大小，通过batch接口指定，并设置是否丢弃无法被batch size整除的剩余数据。

> 调用数据集的`map`、`split`、`batch`为数据集处理流水线增加对应操作，返回值为新的Dataset类型。现在仅定义流水线操作，在执行时开始执行数据处理流水线，获取最终处理好的数据并送入模型进行训练。

imdb_train = imdb_train.batch(64, drop_remainder=True)
    imdb_valid = imdb_valid.batch(64, drop_remainder=True)

![在这里插入图片描述][9c27a7354961400bb0874ea413a008b3.png]

## 四、模型构建 ##

完成数据集的处理后，我们设计用于情感分类的模型结构。首先需要将输入文本(即序列化后的index id列表)通过查表转为向量化表示，此时需要使用`nn.Embedding`层加载Glove词向量；然后使用RNN循环神经网络做特征提取；最后将RNN连接至一个全连接层，即`nn.Dense`，将特征转化为与分类数量相同的size，用于后续进行模型优化训练。整体模型结构如下：

nn.Embedding -> nn.RNN -> nn.Dense

这里我们使用能够一定程度规避RNN梯度消失问题的变种LSTM(Long short term memory)做特征提取层。下面对模型进行详解：

### 1.Embedding ###

Embedding层又可称为EmbeddingLookup层，其作用是使用index id对权重矩阵对应id的向量进行查找，当输入为一个由index id组成的序列时，则查找并返回一个相同长度的矩阵，例如：

embedding = nn.Embedding(1000, 100) # 词表大小(index的取值范围)为1000，表示向量的size为100
    input shape: (1, 16)                # 序列长度为16
    output shape：(1, 16, 100)

这里我们使用前文处理好的Glove词向量矩阵，设置`nn.Embedding`的`embedding_table`为预训练词向量矩阵。对应的`vocab_size`为词表大小400002，`embedding_size`为选用的glove.6B.100d向量大小，即100。

### 2.RNN(循环神经网络) ###

> ![这里是引用][a6dc8f1d62c84e4fa5d66b65a76b7d26.png]  
> ![在这里插入图片描述][8cdd7cee9ed94f769e7d95014cf54506.png]
> 
> ![在这里插入图片描述][7de3c205fb9748b5b04986ed20e41d97.png]  
> ![在这里插入图片描述][2d880ab4848c405ebd2166a4e1e4d02b.png]

### 3.Dense ###

在经过LSTM编码获取句子特征后，将其送入一个全连接层，即`nn.Dense`，将特征维度变换为二分类所需的维度1，经过Dense层后的输出即为模型预测结果。

> 在Dense层后进行了`sigmoid`运算，将预测值归一至`[0,1]`区间，用于后续和`BCELoss`(BinaryCrossEntropyLoss)计算二分类交叉熵损失使用。

import math
    import mindspore as ms
    import mindspore.nn as nn
    import mindspore.numpy as mnp
    import mindspore.ops as ops
    from mindspore.common.initializer import Uniform, HeUniform
    
    class RNN(nn.Cell):
        def __init__(self, embeddings, hidden_dim, output_dim, n_layers,
                     bidirectional, dropout, pad_idx):
            super().__init__()
            vocab_size, embedding_dim = embeddings.shape
            self.embedding = nn.Embedding(vocab_size, embedding_dim, embedding_table=ms.Tensor(embeddings), padding_idx=pad_idx)
            self.rnn = nn.LSTM(embedding_dim,
                               hidden_dim,
                               num_layers=n_layers,
                               bidirectional=bidirectional,
                               dropout=dropout,
                               batch_first=True)
            weight_init = HeUniform(math.sqrt(5))
            bias_init = Uniform(1 / math.sqrt(hidden_dim * 2))
            self.fc = nn.Dense(hidden_dim * 2, output_dim, weight_init=weight_init, bias_init=bias_init)
            self.dropout = nn.Dropout(1 - dropout)
            self.sigmoid = ops.Sigmoid()
    
        def construct(self, inputs):
            embedded = self.dropout(self.embedding(inputs))
            _, (hidden, _) = self.rnn(embedded)
            hidden = self.dropout(mnp.concatenate((hidden[-2, :, :], hidden[-1, :, :]), axis=1))
            output = self.fc(hidden)
            return self.sigmoid(output)

![在这里插入图片描述][eedf78f05a7c4a608311c021dab36600.png]

### 4.损失函数与优化器 ###

完成模型主体构建后，首先根据指定的参数实例化网络；然后选择损失函数和优化器，并将其使用nn.TrainOneStepCell进行封装。针对本节情感分类问题的特性，即预测Positive或Negative的二分类问题，我们选择nn.BCELoss(二分类交叉熵损失函数)，这里也可以选择`nn.BCEWithLogitsLoss`, 其包含`sigmoid`运算，即：

BCEWithLogitsLoss = Sigmoid + BCELoss

这里使用BECLoss需要设置`reduction`参数为均值。而后使用`nn.WithLossCell`将其与实例化网络对象进行关联。

选择合适的损失函数后，选择Adam优化器，并将二者传入`TrainOneStepCell`中。

> MindSpore的设计理念为整图计算与优化，因此将Loss  
> function和Optimizer均视为计算图的一部分，所以构造`TrainOneStepCell`作为Wrapper使用。

hidden_size = 256
    output_size = 1
    num_layers = 2
    bidirectional = True
    dropout = 0.5
    lr = 0.001
    pad_idx = vocab.tokens_to_ids('<pad>')
    
    net = RNN(embeddings, hidden_size, output_size, num_layers, bidirectional, dropout, pad_idx)
    loss = nn.BCELoss(reduction='mean')
    net_with_loss = nn.WithLossCell(net, loss)
    optimizer = nn.Adam(net.trainable_params(), learning_rate=lr)
    train_one_step = nn.TrainOneStepCell(net_with_loss, optimizer)

![在这里插入图片描述][75f6a78fa69e4613a178abe3586037e7.png]

### 5.训练逻辑 ###

在完成模型构建，进行训练逻辑的设计。一般训练逻辑分为一下步骤：

1.  读取一个Batch的数据；
2.  送入网络，进行正向计算和反向传播，更新权重；
3.  返回loss。

下面按照此逻辑，使用`tqdm`库，设计训练一个epoch的函数，用于训练过程和loss的可视化。

def train_one_epoch(model, train_dataset, epoch=0):
        model.set_train()
        total = train_dataset.get_dataset_size()
        loss_total = 0
        step_total = 0
        with tqdm(total=total) as t:
            t.set_description('Epoch %i' % epoch)
            for i in train_dataset.create_tuple_iterator():
                loss = model(*i)
                loss_total += loss.asnumpy()
                step_total += 1
                t.set_postfix(loss=loss_total/step_total)
                t.update(1)

![在这里插入图片描述][1d8e078a211548bdb7ed29d9a15b7da2.png]

### 6.评估指标和逻辑 ###

训练逻辑完成后，需要对模型进行评估。即使用模型的预测结果和测试集的正确标签进行对比，求出预测的准确率。由于IMDB的情感分类为二分类问题，对预测值直接进行四舍五入即可获得分类标签(0或1)，然后判断是否与正确标签相等即可。下面为二分类准确率计算函数实现：

def binary_accuracy(preds, y):
        """
        计算每个batch的准确率
        """
    
        # 对预测值进行四舍五入
        rounded_preds = np.around(preds)
        correct = (rounded_preds == y).astype(np.float32)
        acc = correct.sum() / len(correct)
        return acc

![在这里插入图片描述][852f642a3b5d4748a0f0f8d7ddb782cc.png]

有了准确率计算函数后，类似于训练逻辑，对评估逻辑进行设计, 分别为以下步骤：

1.  读取一个Batch的数据；
2.  送入网络，进行正向计算，获得预测结果；
3.  计算准确率。

同训练逻辑一样，使用`tqdm`进行loss和过程的可视化。此外返回评估loss至供保存模型时作为模型优劣的判断依据。

> 在进行evaluate时，使用的模型是不包含损失函数和优化器的网络主体；  
> 在进行evaluate前，需要通过`model.set_train(False)`将模型置为评估状态，此时Dropout不生效。

def evaluate(model, test_dataset, criterion, epoch=0):
        total = test_dataset.get_dataset_size()
        epoch_loss = 0
        epoch_acc = 0
        step_total = 0
        model.set_train(False)
    
        with tqdm(total=total) as t:
            t.set_description('Epoch %i' % epoch)
            for i in test_dataset.create_tuple_iterator():
                predictions = model(i[0])
                loss = criterion(predictions, i[1])
                epoch_loss += loss.asnumpy()
    
                acc = binary_accuracy(predictions.asnumpy(), i[1].asnumpy())
                epoch_acc += acc
    
                step_total += 1
                t.set_postfix(loss=epoch_loss/step_total, acc=epoch_acc/step_total)
                t.update(1)
    
        return epoch_loss / total

![在这里插入图片描述][962f062ed4ae436e8128d41119d3ea83.png]

## 五、模型训练与保存 ##

前序完成了模型构建和训练、评估逻辑的设计，下面进行模型训练。这里我们设置训练轮数为5轮。同时维护一个用于保存最优模型的变量`best_valid_loss`，根据每一轮评估的loss值，取loss值最小的轮次，将模型进行保存。

MindSpore默认采用静态图模式(即Define and Run)进行训练，第一个step会进行计算图编译操作，较为耗时，但整体训练效率更高，若需要进行单步调试或使用动态图模式，可使用如下代码进行设置：

import mindspore as ms
    ms.set_context(mode=ms.PYNATIVE_MODE)

![在这里插入图片描述][f06ff94e13c441c38e3f4a87bca681f4.png]

num_epochs = 5
    best_valid_loss = float('inf')
    ckpt_file_name = os.path.join(cache_dir, 'sentiment-analysis.ckpt')
    
    for epoch in range(num_epochs):
        train_one_epoch(train_one_step, imdb_train, epoch)
        valid_loss = evaluate(net, imdb_valid, loss, epoch)
    
        if valid_loss < best_valid_loss:
            best_valid_loss = valid_loss
            ms.save_checkpoint(net, ckpt_file_name)

![在这里插入图片描述][c5ae35c6c99441b49053dd3ea892d135.png]  
![在这里插入图片描述][50260ef869fc4c2f8417dab47be202f6.png]

## 六、模型加载与测试 ##

模型训练完成后，一般需要对模型进行测试或部署上线，此时需要加载已保存的最优模型(即checkpoint)，供后续测试使用。这里我们直接使用MindSpore提供的Checkpoint加载和网络权重加载接口：1.将保存的模型Checkpoint加载到内存中，2.将Checkpoint加载至模型。

> load\_param\_into\_net接口会返回模型中没有和Checkpoint匹配的权重名，正确匹配时返回空列表。

param_dict = ms.load_checkpoint(ckpt_file_name)
    ms.load_param_into_net(net, param_dict)

![在这里插入图片描述][2ee6d4f568ae45638e40411a9b28427d.png]

对测试集打batch，然后使用evaluate方法进行评估，得到模型在测试集上的效果。

imdb_test = imdb_test.batch(64)
    evaluate(net, imdb_test, loss)

![在这里插入图片描述][9c0a968d68ea4f34ae2821c208496045.png]  
![在这里插入图片描述][46612fac9fcb4c97ad267b63aacefe66.png]

## 七、自定义输入测试 ##

最后我们设计一个预测函数，实现开头描述的效果，输入一句评价，获得评价的情感分类。具体包含以下步骤:

1.  将输入句子进行分词；
2.  使用词表获取对应的index id序列；
3.  index id序列转为Tensor；
4.  送入模型获得预测结果；
5.  打印输出预测结果。

具体实现如下：

score_map = {
        
        1: "Positive",
        0: "Negative"
    }
    
    def predict_sentiment(model, vocab, sentence):
        model.set_train(False)
        tokenized = sentence.lower().split()
        indexed = vocab.tokens_to_ids(tokenized)
        tensor = ms.Tensor(indexed, ms.int32)
        tensor = tensor.expand_dims(0)
        prediction = model(tensor)
        return score_map[int(np.round(prediction.asnumpy()))]

![在这里插入图片描述][6e34827f863d47a6b0fe0889b5e0f78b.png]

predict_sentiment(net, vocab, "This film is great")

![在这里插入图片描述][751dcc5a5ba0467fb3657f1220bb0d96.png]

[MindSpore]: https://bbs.csdn.net/forums/MindSpore_Official
[a9571b75523b4cbca146948dbadf8715.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/4e6a768cd96849a9a5a0081bc9d5f2d6.png
[ModelArts]: https://console.huaweicloud.com/modelarts/?region=cn-north-4#/dashboard
[ee2578b7b26f4435bfd12aedd8959967.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/9e8f6ba2b81b4d608111ffe4a73a5bb2.png
[547900ad31bf4f4386132554fe26dfb9.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/f286d54b266a4510a2bd390005f2b424.png
[1466f73e44fa4fe69bd38b0b30661e22.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/794dd5e6cd5b453fb8b4c44f10e9d067.png
[NoteBook]: https://obs.dualstack.cn-north-4.myhuaweicloud.com/mindspore-website/notebook/r1.9/tutorials/application/zh_cn/nlp/mindspore_sentiment_analysis.py
[19d112a9f67e4b48ae558cc0a4a8ce1e.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/03db8fbcfab34c2b99c3ddb3bbf303de.png
[e61bd133ba3542f09c446844d843b8fb.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/d0643cdb6f7846a197f8cd46aa89e297.png
[5979276a6ed045c580aa4f40b2209f4d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/d809fcffbae9410a8636010267fe97c0.png
[a59b36db7dac45079c8d194c1a8b504b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/3804cc8fccd64b8e9a257e6dab3af05f.png
[550d87512bab4b3a9c4d13138c32769d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/8decaf198a9245e3968572ccc33f0a76.png
[MindSpore 1]: https://www.mindspore.cn/
[2dec09846d164105b79d514aa0deed82.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/c347d7b134ba4f45b2a5d7db9fcadb06.png
[744a4ea99b0b4cba82dbbfd9f290e402.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/40e3d0ea0ed840c0b4fc9f11d23ea4af.png
[d1b5e7f5355b410599f32ed6a59d9000.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/5f50d52ff1a74620b153c7695c30159d.png
[7eec4a2d97634beeae03da93eabf8eb5.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/ed46caea3d2141d08e06e58d0d2e5710.png
[62986501180f4c44a9cbda230df95070.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/f4273b0058fc49b5add5402423e01d91.png
[IMDB]: https://ai.stanford.edu/~amaas/data/sentiment/
[Glove]: https://nlp.stanford.edu/projects/glove/
[e5856d8803a540f1a60701cacdebc8fa.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/a3a38949701f4eaba8259ce4ac0f0cca.png
[f76f5b6dd26e46d5bb7fec945173db5a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/ea2195f46a7744b38a4328af5c44b46e.png
[81256546a2d34dc3a29bafb8273154b4.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/05397d8d3c764e5292d8b31f45e75cf4.png
[91116e702e9e4364871b442c0a4da1cf.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/c553ff1de4844dd790bb7f45377ad9cc.png
[6bb202235b0347319c1e421ac9b95426.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/b2e80dfd7ce54f74acc1569fe902bc2b.png
[14a0e215164545c3b3b80697ea1aae46.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/a12bdd7a9f414bee9631d965cfe50bf3.png
[c2f5f0ed47b44991af0aeb071300bdd6.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/b0b52dc6a8c041bf8282460ef429e556.png
[c4814c22dd2346c1937e47a3b427a130.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/bfb832ff7bae4cfebbadcd3e5731eefb.png
[750538bed2b74bab80e12264d092ac69.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/60453906bacf491cb50c404351c628fb.png
[c815d283678945a1bf18affe273a6b02.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/3a52d3fe80194f048350218a304a8b9d.png
[MindSpore 2]: https://www.mindspore.cn/docs/zh-CN/r1.9/design/data_engine.html
[68ed4083438e4e328227a25cd22f206d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/c39e0b50b7e841cdb0c4b670b2f4f8d8.png
[deb57adf87924a10be7dc309095682c2.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/dab766d6ddd848829f1a2c644332d7ec.png
[c9af93b4358140caa6da5988109d0c6a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/b0c918ce198343ed9b1e54f6d2724cf5.png
[9c27a7354961400bb0874ea413a008b3.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/836b8f7ba805439fa2f661e8ca07af09.png
[a6dc8f1d62c84e4fa5d66b65a76b7d26.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/95a0123183fc4180b0b5061c76065f91.png
[8cdd7cee9ed94f769e7d95014cf54506.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/ef9ed09d4dbb4f0986e93323731cf8f6.png
[7de3c205fb9748b5b04986ed20e41d97.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/2992f60f684e48da9e94596e42281920.png
[2d880ab4848c405ebd2166a4e1e4d02b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/b255a44fafdf411d915d2e924aafa885.png
[eedf78f05a7c4a608311c021dab36600.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/a22a716c0f6e453a9f787514cc5d0445.png
[75f6a78fa69e4613a178abe3586037e7.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/8fbb885c8e4944a486b75a4e14853550.png
[1d8e078a211548bdb7ed29d9a15b7da2.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/3f1e88e13fa44319a99c855f9c73cb69.png
[852f642a3b5d4748a0f0f8d7ddb782cc.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/e34b5894affe4e5c8bf5d341d77464bd.png
[962f062ed4ae436e8128d41119d3ea83.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/f1a8850d06a34e34bd7a75161711553b.png
[f06ff94e13c441c38e3f4a87bca681f4.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/e99daf2b639642e7a0e8485231e5c6a0.png
[c5ae35c6c99441b49053dd3ea892d135.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/74ed2371335246c18c4f72928b96a9c0.png
[50260ef869fc4c2f8417dab47be202f6.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/cc84892e9ae04cf9a94b6b74eb92fe39.png
[2ee6d4f568ae45638e40411a9b28427d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/3246b178b1354f50855d11cde59c17e3.png
[9c0a968d68ea4f34ae2821c208496045.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/10bffa0793bb4f419ce3fcb40c2908f9.png
[46612fac9fcb4c97ad267b63aacefe66.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/6f5d3a19f1914b4aaf2b5e14634f019f.png
[6e34827f863d47a6b0fe0889b5e0f78b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/2fdd59f7683a407c803ec2611469b055.png
[751dcc5a5ba0467fb3657f1220bb0d96.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/30/06c7721789fb47faa31332aafdbfca1d.png