视频加载失败

课程

713 字
约 3 分钟

实验三 Skip-gram 词向量实验

自然语言处理labs/lab/lab03·更新于 2026-09-15

实验三 Skip-gram 词向量实验

目录说明

  • lab03_skipgram.py:实验3的主脚本,负责数据准备、模型训练、验证损失统计和词语余弦相似度计算。
  • data/skipgram_corpus.txt:默认训练语料。为了对应题目中的英文词对示例,这里提供了一份可直接运行的小型英文语料。
  • outputs/:脚本运行后生成的配置、词表、训练指标和相似度结果。
  • docs/lab03_report.md:实验分析、题目问答和运行说明。

依赖安装

建议在工作区虚拟环境中执行:

f:/Study/ComputerScience/Computer_2026_Spring_Term/Natural_Language_Processing/lab/.venv/Scripts/python.exe -m pip install -r lab03/requirements.txt

默认只要求 CPU 可运行。若本机安装的是支持 CUDA 的 Paddle 版本,可以在运行时传入 --device gpu

运行命令

f:/Study/ComputerScience/Computer_2026_Spring_Term/Natural_Language_Processing/lab/.venv/Scripts/python.exe lab03/lab03_skipgram.py

常用可调参数:

f:/Study/ComputerScience/Computer_2026_Spring_Term/Natural_Language_Processing/lab/.venv/Scripts/python.exe lab03/lab03_skipgram.py --epochs 80 --embedding-dim 48 --window-size 3 --device cpu

最小验证点

  • 终端应先打印加载到的 token 数、词表大小和训练样本数。
  • 训练过程中应持续打印 train_loss,若存在验证集还会打印 val_loss
  • 训练结束后应输出 5 组词对的余弦相似度。
  • outputs/ 目录下应生成 run_config.jsonvocab.jsontraining_metrics.jsonsimilarity_results.json

默认语料选择说明

题目示例中的词对是英文词汇,如 king/queentopic/theme。为了保证仓库内可以直接复现实验流程,默认语料选择了一份小型英文合成语料,把这些目标词放在可重复的上下文里,便于在短时间训练后观察到合理的相似度趋势。

Profile Image of the Author
Sonder
好想要技术
这是公告标题
这只是一个公告
分类
标签
站点信息
构建平台
GitHub Actions
博客版本
Firefly v6.16.7
文章许可
CC BY-NC-SA 4.0
文章目录