RIPGeo代码理解（七）test.py（加载检查点，然后测试）

​代码链接：RIPGeo代码实现

    ├── preprocess.py # 预处理数据集并为模型运行执行IP聚类
    ├── main.py # 运行模型进行训练和测试
    ├── test.py #加载检查点，然后测试

 一、导入各种模块和数据库

import torch.nn

from lib.utils import *
import argparse
import numpy as np
import random
from lib.model import *
import copy
import pandas as pd

整体功能是准备运行一个 PyTorch 深度学习模型的环境，具体的功能实现需要查看 lib.utils、lib.model 中的代码，以及整个文件的后续部分。

1、from lib.utils import *：从 lib.utils 模块中导入所有内容。

2、from lib.model import *：从 lib.model 模块中导入所有内容。

3、import copy：导入 copy 模块，用于复制对象，通常用于创建对象的深拷贝。

二、参数初始化（通过命令行参数）

parser = argparse.ArgumentParser()
# parameters of initializing
parser.add_argument('--seed', type=int, default=2022, help='manual seed')
parser.add_argument('--model_name', type=str, default='RIPGeo')
parser.add_argument('--dataset', type=str, default='New_York', choices=["Shanghai", "New_York", "Los_Angeles"],
                    help='which dataset to use')

这部分代码的目的是通过命令行参数设置一些初始化的参数，例如随机数种子、模型名称和数据集名称。这使得在运行脚本时可以通过命令行参数来指定这些参数的值。

1、parser = argparse.ArgumentParser()：创建一个 argparse.ArgumentParser 对象，用于解析命令行参数。

2、parser.add_argument('--seed', type=int, default=1024, help='manual seed')：添加一个命令行参数，名称为 '--seed'，表示随机数种子，类型为整数，默认值为 2022，help 参数是在命令行中输入 --help 时显示的帮助信息。

3、parser.add_argument('--model_name', type=str, default='RIPGeo')：添加一个命令行参数，名称为 '--model_name'，表示模型的名称，类型为字符串，默认值为 'RIPGeo'。

4、parser.add_argument('--dataset', type=str, default='New_York', choices=["Shanghai", "New_York", "Los_Angeles"], help='which dataset to use')：添加一个命令行参数，名称为 '--dataset'，表示数据集的名称，类型为字符串，默认值为 'New_York'，choices 参数指定了可选的值为 ["Shanghai", "New_York", "Los_Angeles"]，用户只能从这三个值中选择。

三、训练过程参数设置

# parameters of training
parser.add_argument('--beta1', type=float, default=0.9)
parser.add_argument('--beta2', type=float, default=0.999)

parser.add_argument('--lr', type=float, default=2e-3)
parser.add_argument('--harved_epoch', type=int, default=20)
parser.add_argument('--early_stop_epoch', type=int, default=200)
parser.add_argument('--saved_epoch', type=int, default=200)
parser.add_argument('--load_epoch', type=int, default=200)
parser.add_argument('--dropout', type=float, default=0.1)

这部分代码的目的是设置一些训练过程中的超参数，例如优化器的动量参数、学习率、权重参数等。这些参数在训练过程中会影响模型的更新和收敛速度。

1、parser.add_argument('--beta1', type=float, default=0.9)：添加一个命令行参数，名称为 '--beta1'，表示 Adam 优化器的第一个动量（momentum）参数，类型为浮点数，默认值为 0.9。

2、parser.add_argument('--beta2', type=float, default=0.999)：添加一个命令行参数，名称为 '--beta2'，表示 Adam 优化器的第二个动量参数，类型为浮点数，默认值为 0.999。

3、parser.add_argument('--lr', type=float, default=2e-3)：添加一个命令行参数，名称为 '--lr'，表示学习率，类型为浮点数，默认值为 2e-3。

4、parser.add_argument('--harved_epoch', type=int, default=20)：添加一个命令行参数，名称为 '--harved_epoch'，表示当连续多少个epoch的性能没有增加时，学习率减半，类型为整数，默认值为 20。

5、parser.add_argument('--early_stop_epoch', type=int, default=200)：添加一个命令行参数，名称为 '--early_stop_epoch'，表示当连续多少个epoch的性能没有增加时，训练停止，类型为整数，默认值为 200。

6、parser.add_argument('--saved_epoch', type=int, default=200)：  添加一个命令行参数，名称为 '--saved_epoch'，表示为测试保存多少个checkpoint（epoch），类型为整数，默认值为 200。

7、parser.add_argument('--load_epoch', type=int, default=200)：添加一个命令行参数，名称为 '--load_epoch'，