# chinese-text-classification

**Repository Path**: hl0929/chinese-text-classification

## Basic Information

- **Project Name**: chinese-text-classification
- **Description**: 基于 Pytorch 的中文文本分类代码库
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2023-08-30
- **Last Updated**: 2024-03-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README


# 项目介绍

一行命令完成中文文本分类模型训练、**部署**。

* 一键启动
```bash
nohup bash launch.sh > train.log 2>&1 &
```

* 体验服务
```bash
curl --location 'http://localhost:8001/cls/' --header 'Content-Type: application/json' --data '{"query": "姚明是谁"}'
```

* 响应结果
```bash
{
	"predict": "sports",
	"elapsed": "0.0007787439972162247"
}
```

# 项目环境

* 虚拟环境
```bash
python3 -m venv venv

source venv/bin/activate
```

* 安装依赖
```bash
pip install -r requirements.txt
```


# 执行脚本

* 训练
```bash
bash ./train/run.sh 
```

* 推理
```bash
bash ./inference/transform.sh
```

* 部署
```bash
bash ./service/deploy.sh
```

* 请求
```bash
curl --location 'http://localhost:8000/cls/' --header 'Content-Type: application/json' --data '{"query": "姚明是谁"}'
```

# 单元测试

* 运行所有测试文件
```bash
nosetests -v tests/*
```

* 运行具体某一个测试文件
```bash
python -m unittest -v tests.test_text_cnn
```


# 原始数据

[THUCNews](http://thuctc.thunlp.org/)


# 模型实现

[text_classification_AI100](https://github.com/lc222/text_classification_AI100)

[Chinese-Text-Classification-Pytorch](https://github.com/649453932/Chinese-Text-Classification-Pytorch)

[ChineseTextClassifier](https://github.com/ami66/ChineseTextClassifier)

[Keras-TextClassification](https://github.com/yongzhuo/Keras-TextClassification)

[bert-Chinese-classification-task](https://github.com/NLPScott/bert-Chinese-classification-task)

[text-classification](https://github.com/wavewangyue/text-classification)

[text_classification](https://github.com/brightmart/text_classification)


# TensorBoard 可视化

执行如下命令：

```bash
 tensorboard --logdir=./logdir  # ./logdir 日志文件所在目录
 ```