剧透

本章内容概要

在本章中，我们将：

• 定义一个用于处理模型训练的类。

• 实现类的构造函数方法。

• 理解类中公有、受保护和私有方法之间的区别。

• 将我们迄今为止开发的代码整合到该类中。

• 实例化这个类，并用它运行一个优雅的完整流程。

导入依赖

为了保持结构清晰，本章所需的所有库都将在开头统一导入。本章需要以下导入：

1import numpy as np
2import datetime
3
4import torch
5import torch.optim as optim
6import torch.nn as nn
7import torch.functional as F
8from torch.utils.data import DataLoader, TensorDataset, random_split
9from torch.utils.tensorboard import SummaryWriter
10
11import matplotlib.pyplot as plt
12%matplotlib inline
13plt.style.use('fivethirtyeight')

Jupyter Notebook

本章结束时，你将获得包含全部代码的 Jupyter Notebook。你可以运行完整代码并进行实验。