Chinese-resume-classification

本项目目的:对高校教师的简历做分类,预测其专业领域。

所使用方法:Word2Vec预训练词向量模型 + TextCNN文本分类模型。

本项目的代码实现完全通过python实现,其中Word2Vec使用gensim包实现,TextCNN的pytorch实现参考https://github.com/Shawn1993/cnn-text-classification-pytorch。

项目介绍博客:https://blog.csdn.net/Jeff_zjf/article/details/107535329

环境安装

pip install gensim

pip install torch torchtext # 不同平台上安装脚本可能不同,需查询https://pytorch.org/

pip install word2vec

pip install xlrd>=0.9.0

pip install bs4

下载数据模型

百度云盘:https://pan.baidu.com/s/1Z88bmTh14eptERjmY7DmDg,提取码:4kwi

下载完成后,把 snapshot.zip 解压到 Chinese-resume-classification/ 下

把 data_.zip 解压到 Chinese-resume-classification/data/ 下

把 model.zip 解压到 Chinese-resume-classification/word2Vec/ 下

预测专业类别

直接通过简历链接,自动爬取文本,预测其专业类别。

可以以脚本形式运行:

python main.py -use-word2vec -predict-url https://baike.baidu.com/item/%E4%BD%95%E6%81%BA%E6%98%8E/22863446?fr=aladdin

这里的链接可以替换成你想要预测的简历链接。

或者使用GUI界面:

python demo.py

模型训练

python main.py -use-word2vec