课程进度 28% · 第3/8章第3/8章 · 标签 1/4
— 1 —
代码规范
PEP 8规范
PEP 8是Python代码的官方风格指南,包括缩进(4空格)、行宽(79字符)、命名约定等。
python
1
# 正确的命名
2
class ImageClassifier: # 类名驼峰
3
def predict(self): pass # 方法名小写
4
MAX_ITER = 1000 # 常量全大写
5
6
def load_data(file_path): pass # 函数名小写+下划线
类型注解
类型注解提高代码可读性,配合mypy进行静态类型检查。
— 2 —
AI代码规范
AI项目特有的编程规范,包括随机种子设置、模型保存、日志记录等。
python
1
import random, numpy as np, torch
2
3
def set_seed(seed=42):
4
random.seed(seed)
5
np.random.seed(seed)
6
torch.manual_seed(seed)
7
torch.cuda.manual_seed_all(seed)
8
9
# 训练循环规范
10
for epoch in range(epochs):
11
model.train()
12
for batch in loader:
13
loss = train_step(model, batch)
14
logger.log(loss) # 使用logger而非print