Skip to content

Chinese-Text-Classification-Pytorch 的推理 server 项目,需要 分类、词表、模型类、模型权重。

Notifications You must be signed in to change notification settings

AriesYB/keyword_classifier

Repository files navigation

品目分类器

此项目只推理,训练项目

一、配置环境

Python 版本3.12.4

1.创建虚拟环境

python -m venv venv

2.激活虚拟环境

Linux 命令: source venv/bin/activate

Windows 命令: venv\bin\activate.bat

3.安装依赖

pip install -r requirements.txt

cpu 版本的pytorch pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu

gpu 版本的pytorch pip install torch==2.4.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html

二、使用

1.启动 server

python classifier_server.py

2.预测关键词

http://127.0.0.1:5000/predict?keyword=联想笔记本电脑

About

Chinese-Text-Classification-Pytorch 的推理 server 项目,需要 分类、词表、模型类、模型权重。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages