Python 机器学习实战：模型构建与调优基础

进阶框架实现

在实际项目中，我们更多依赖成熟的深度学习框架。以下是基于 TensorFlow 和 PyTorch 的标准实现模式。

TensorFlow 版本

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

class TensorFlowModel:
    """TensorFlow 版本的模型实现"""
    def __init__(self, input_dim: int, hidden_units: List[int] = [64, 32]):
        """初始化 TensorFlow 模型
        
        Args:
            input_dim: 输入维度
            hidden_units: 隐藏层单元数列表
        """
        self.model = self._build_model(input_dim, hidden_units)

    def _build_model(self, input_dim: int, hidden_units: List[int]) -> keras.Model:
        """构建模型架构"""
        inputs = keras.Input(shape=(input_dim,))
        x = inputs
        for units in hidden_units:
            x = layers.Dense(units, activation='relu')(x)
            x = layers.BatchNormalization()(x)
            x = layers.Dropout(0.2)(x)
        outputs = layers.Dense(1)(x)
        model = keras.Model(inputs=inputs, outputs=outputs)
        model.compile(
            optimizer=keras.optimizers.Adam(learning_rate=0.001),
            loss='mse',
            metrics=['mae'])
        return model

    def train(self, X_train, y_train, X_val, y_val, epochs=100, batch_size=32):
        """训练模型"""
        history = self.model.fit(
            X_train, y_train,
            validation_data=(X_val, y_val),
            epochs=epochs,
            batch_size=batch_size,
            verbose=1)
        return history

    def predict(self, X):
        """预测"""
        return self.model.predict(X)

PyTorch 版本

import torch
import torch.nn as nn
import torch.optim as optim

class PyTorchModel(nn.Module):
    """PyTorch 版本的模型实现"""
    def __init__(self, input_dim: int, hidden_units: List[int] = [64, 32]):
        """初始化 PyTorch 模型
        
        Args:
            input_dim: 输入维度
            hidden_units: 隐藏层单元数列表
        """
        super(PyTorchModel, self).__init__()
        layers_list = []
        prev_units = input_dim
        for units in hidden_units:
            layers_list.append(nn.Linear(prev_units, units))
            layers_list.append(nn.ReLU())
            layers_list.append(nn.BatchNorm1d(units))
            layers_list.append(nn.Dropout(0.2))
            prev_units = units
        layers_list.append(nn.Linear(prev_units, 1))
        self.network = nn.Sequential(*layers_list)

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        """前向传播"""
        return self.network(x)

    def train_model(self, train_loader, val_loader, epochs=100, lr=0.001):
        """训练模型"""
        criterion = nn.MSELoss()
        optimizer = optim.Adam(self.parameters(), lr=lr)
        train_losses = []
        val_losses = []

        for epoch in range(epochs):
            # 训练阶段
            self.train()
            train_loss = 0.0
            for X_batch, y_batch in train_loader:
                optimizer.zero_grad()
                outputs = self(X_batch)
                loss = criterion(outputs, y_batch)
                loss.backward()
                optimizer.step()
                train_loss += loss.item()

            # 验证阶段
            self.eval()
            val_loss = 0.0
            with torch.no_grad():
                for X_batch, y_batch in val_loader:
                    outputs = self(X_batch)
                    loss = criterion(outputs, y_batch)
                    val_loss += loss.item()
            
            train_losses.append(train_loss / len(train_loader))
            val_losses.append(val_loss / len(val_loader))
            if (epoch + 1) % 10 == 0:
                print(f"Epoch {epoch+1}/{epochs}, "
                      f"Train Loss: {train_losses[-1]:.4f}, "
                      f"Val Loss: {val_losses[-1]:.4f}")
        return train_losses, val_losses

数据处理与评估

完整数据处理流程

数据质量直接决定模型上限。以下是一个标准化的预处理类，涵盖缺失值处理、编码及标准化。

import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler, LabelEncoder
from sklearn.impute import SimpleImputer
from typing import List, Tuple

class DataProcessor:
    """数据处理类"""
    def __init__(self):
        self.scaler = StandardScaler()
        self.label_encoders = {}
        self.imputer = SimpleImputer(strategy='mean')

    def process(self, data: pd.DataFrame, target_col: str, categorical_cols: List[str] = None, test_size: float = 0.2) -> Tuple:
        """完整的数据处理流程
        
        Args:
            data: 原始数据
            target_col: 目标列名
            categorical_cols: 类别列名列表
            test_size: 测试集比例
        Returns:
            处理后的训练集和测试集
        """
        # 1. 分离特征和目标
        X = data.drop(columns=[target_col])
        y = data[target_col]

        # 2. 处理缺失值
        X = pd.DataFrame(
            self.imputer.fit_transform(X.select_dtypes(include=[np.number])),
            columns=X.select_dtypes(include=[np.number]).columns
        )

        # 3. 编码类别特征
        if categorical_cols:
            for col in categorical_cols:
                if col in X.columns:
                    le = LabelEncoder()
                    X[col] = le.fit_transform(X[col].astype(str))
                    self.label_encoders[col] = le

        # 4. 标准化
        X_scaled = self.scaler.fit_transform(X)

        # 5. 划分数据集
        X_train, X_test, y_train, y_test = train_test_split(
            X_scaled, y, test_size=test_size, random_state=42)
        return X_train, X_test, y_train, y_test

# 使用示例
if __name__ == "__main__":
    data = pd.DataFrame({
        'feature1': np.random.randn(1000),
        'feature2': np.random.randn(1000),
        'feature3': np.random.choice(['A', 'B', 'C'], 1000),
        'target': np.random.randn(1000)
    })
    processor = DataProcessor()
    X_train, X_test, y_train, y_test = processor.process(
        data, target_col='target', categorical_cols=['feature3'])
    print(f"训练集形状：{X_train.shape}")
    print(f"测试集形状：{X_test.shape}")

模型评估工具

科学的评估指标能帮助我们客观判断模型表现。分类任务关注准确率、召回率等，回归任务则侧重 MSE、R²等。

from sklearn.metrics import (
    accuracy_score, precision_score, recall_score,
    f1_score, roc_auc_score, confusion_matrix,
    classification_report, mean_squared_error,
    mean_absolute_error, r2_score
)
import matplotlib.pyplot as plt
import seaborn as sns

class ModelEvaluator:
    """模型评估类"""
    @staticmethod
    def evaluate_classification(y_true, y_pred, y_prob=None):
        """评估分类模型"""
        metrics = {
            'accuracy': accuracy_score(y_true, y_pred),
            'precision': precision_score(y_true, y_pred, average='weighted'),
            'recall': recall_score(y_true, y_pred, average='weighted'),
            'f1': f1_score(y_true, y_pred, average='weighted')
        }
        if y_prob is not None:
            metrics['roc_auc'] = roc_auc_score(y_true, y_prob, multi_class='ovr')
        return metrics

    @staticmethod
    def evaluate_regression(y_true, y_pred):
        """评估回归模型"""
        return {
            'mse': mean_squared_error(y_true, y_pred),
            'rmse': np.sqrt(mean_squared_error(y_true, y_pred)),
            'mae': mean_absolute_error(y_true, y_pred),
            'r2': r2_score(y_true, y_pred)
        }

    @staticmethod
    def plot_confusion_matrix(y_true, y_pred, labels=None):
        """绘制混淆矩阵"""
        cm = confusion_matrix(y_true, y_pred)
        plt.figure(figsize=(8, 6))
        sns.heatmap(cm, annot=True, fmt='d', cmap='Blues',
                    xticklabels=labels, yticklabels=labels)
        plt.title('混淆矩阵')
        plt.xlabel('预测值')
        plt.ylabel('真实值')
        plt.show()

    @staticmethod
    def plot_learning_curve(train_losses, val_losses):
        """绘制学习曲线"""
        plt.figure(figsize=(10, 6))
        plt.plot(train_losses, label='训练损失')
        plt.plot(val_losses, label='验证损失')
        plt.xlabel('Epoch')
        plt.ylabel('Loss')
        plt.title('学习曲线')
        plt.legend()
        plt.grid(True)
        plt.show()

# 使用示例
if __name__ == "__main__":
    # 分类评估示例
    y_true_cls = [0, 1, 0, 1, 0, 1, 0, 0, 1, 1]
    y_pred_cls = [0, 1, 0, 0, 0, 1, 1, 0, 1, 1]
    cls_metrics = ModelEvaluator.evaluate_classification(y_true_cls, y_pred_cls)
    print("分类指标:", cls_metrics)

    # 回归评估示例
    y_true_reg = np.array([1.0, 2.0, 3.0, 4.0, 5.0])
    y_pred_reg = np.array([1.1, 1.9, 3.2, 3.8, 5.1])
    reg_metrics = ModelEvaluator.evaluate_regression(y_true_reg, y_pred_reg)
    print("回归指标:", reg_metrics)

最佳实践与建议

环境准备与项目结构

良好的工程习惯能让项目更易维护。建议采用标准的目录结构，并严格管理依赖。

推荐目录结构：

project/
├── data/          # 数据目录
│   ├── raw/       # 原始数据
│   └── processed/ # 处理后数据
├── notebooks/     # Jupyter 笔记本
├── src/           # 源代码
│   ├── data/      # 数据处理
│   ├── features/  # 特征工程
│   ├── models/    # 模型定义
│   └── utils/     # 工具函数
├── tests/         # 测试代码
├── configs/       # 配置文件
└── requirements.txt # 依赖列表

安装核心库：

conda create -n ai_env python=3.9
conda activate ai_env
pip install numpy pandas matplotlib seaborn scikit-learn tensorflow torch jupyter notebook

常见误区与应对

1. 过拟合问题：若训练集表现优异但测试集差，尝试增加数据量、使用正则化或 Dropout。
2. 数据泄露：确保预处理步骤（如标准化）仅在训练集上拟合，再应用于测试集。
3. 评估偏差：根据任务类型选择合适指标，避免单一指标误导结论。
4. 实验管理：使用版本控制记录代码，保存模型检查点，可视化训练过程以便复现。

总结

本文系统梳理了 Python 机器学习的基础流程。从线性回归的手动实现，到利用 TensorFlow 和 PyTorch 构建深度网络，再到标准化的数据处理与评估体系，这些构成了 AI 开发的基石。实际工作中，建议遵循代码规范，注重实验的可复现性，并根据具体场景灵活调整策略。持续动手实践，将理论转化为解决问题的能力，才是提升技术水平的关键。