python如何训练和保存模型_python进行机器学习的几种模型保存方式!

1、ANN模型

import torch
import torch.nn as nn

# 创建一个示例模型
model = nn.Sequential(
    nn.Linear(10, 5),
    nn.ReLU(),
    nn.Linear(5, 2)
)

# 保存模型到文件
torch.save(model.state_dict(), 'model.pth')

import torch
import torch.nn as nn

# 创建一个与之前示例模型结构相同的实例
model = nn.Sequential(
    nn.Linear(10, 5),
    nn.ReLU(),
    nn.Linear(5, 2)
)

# 加载保存的模型参数
model.load_state_dict(torch.load('model.pth'))

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

2、普通机器学习模型