Fasttext 优化 - 调整训练轮数和学习率

import fasttext


def dm03_epoch():
    """优化手段二：增加训练轮数"""
    # 在上部基础上增加训练轮数（默认 5 轮，改为 25 轮）
    model = fasttext.train_supervised(
        input="data/cooking_pre_train.txt",
        epoch=25,          # 训练轮数：5 → 25
        lr=1.0,            # 学习率
        wordNgrams=2       # n-gram 特征
    )
    
    # 评估
    metrics = model.test("data/cooking_pre_validate.txt")
    print(f"Precision@1: {metrics[1]}")  # 0.42
    
    return model


def dm04_lr():
    """优化手段三：调整学习率"""
    # 在上部基础上调整学习率
    model = fasttext.train_supervised(
        input="data/cooking_pre_train.txt",
        epoch=25,          # 训练轮数
        lr=0.5,            # 学习率：1.0 → 0.5
        wordNgrams=2       # n-gram 特征
    )
    
    # 评估
    metrics = model.test("data/cooking_pre_validate.txt")
    print(f"Precision@1: {metrics[1]}")
    
    return model


# 测试
if __name__ == "__main__":
    print("=== 优化手段二：增加训练轮数 ===")
    model1 = dm03_epoch()
    
    print("\n=== 优化手段三：调整学习率 ===")
    model2 = dm04_lr()