35 lines
807 B
Python
35 lines
807 B
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
主程序入口
|
|
|
|
使用方式：
|
|
|
|
1. 运行单个模型（默认）：
|
|
python main.py
|
|
|
|
修改 config.py 中的 MODEL_TYPE 和 VECTORIZER_TYPE 来切换配置
|
|
|
|
2. 运行对比实验：
|
|
修改 config.py 中 RUN_COMPARISON = True
|
|
|
|
这会依次运行：
|
|
- 实验1: BoW vs TF-IDF (固定LR模型)
|
|
- 实验2: LR vs MLP (固定TF-IDF)
|
|
- 实验3: 不同学习率对比
|
|
- 实验4: 不同隐藏层大小对比
|
|
|
|
最后输出汇总报告
|
|
"""
|
|
|
|
from train import main
|
|
|
|
if __name__ == '__main__':
|
|
print("\n" + "=" * 70)
|
|
print("文本分类实验 - 纯NumPy实现")
|
|
print("数据集: ChnSentiCorp (中文酒店评论)")
|
|
print("模型: Logistic Regression / MLP")
|
|
print("向量化: BoW / TF-IDF")
|
|
print("=" * 70 + "\n")
|
|
|
|
main()
|