KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #typescript #developer-tools #claude-code #ai-agent #codex #deepseek-harness #ai-agents #cli #react #mcp

SentimentAnalysis

★ 32 Open GitHub ↗

中文情感分类 | 基于三分类的文本情感分析

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## 基于三分类的文本情感分析
![GitHub last commit](https://img.shields.io/github/last-commit/yaokui2018/SentimentAnalysis)
![](https://img.shields.io/badge/python-3.7-blue?logo=python&logoColor=FED643)

### 1. 背景介绍

文本情感分析作为NLP的常见任务，具有很高的实际应用价值。本文将采用LSTM模型，训练一个能够识别文本postive, neutral, negative三种情感的分类器。

### 2. LSTM三分类模型

代码需要注意的几点是，第一是，标签需要使用keras.utils.to_categorical来yummy，第二是LSTM二分类的参数设置跟二分有区别，选用softmax，并且loss函数也要改成categorical_crossentropy.

#### 环境依赖
`pip install -r requirements.txt`
#### 训练
`python lstm/lstm_train.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/result.png)
#### 预测
`python lstm/lstm_test.py`

### 3. BiLSTM + Attention
#### 训练
`python bilstm-attention/train.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bilstm-attention.png)
#### 预测
`python bilstm-attention/test.py`

### 4. BiLSTM + Self-Attention
#### 训练
`python bilstm-self-attention/train.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bilstm-self-attention.png)
#### 预测
`python bilstm-self-attention/test.py`

### 5. BiGRU + Self-Attention
#### 训练
`python bigru-self-attention/train.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bigru-self-attention.png)
#### 预测
`python bigru-self-attention/test.py`

### 6. Bert
代码参考自：[liuyijiang1994/bert_senta](https://github.com/liuyijiang1994/bert_senta)
#### 训练
`python bert/train.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bert-train.png)
#### 预测
`python bert/test.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bert.png)
#### 部署
1. 转 onnx

`python bert/torch2onnx.py`

2. onnx 运行

`python bert/deploy.py`

![效果图](https://raw.githubusercontent.com/yaokui2018/SentimentAnalysis/master/imgs/bert-deploy.png)

---

### 说明
代码和数据来源于：[Edward1Chou/SentimentAnalysis](https://github.com/Edward1Chou/SentimentAnalysis)，在基础上做了一些改动。
- 环境由 python2 改为 python3
- 加入 EarlyStopping
- 加入 BiLSTM + Attention -> bilstm-attention/train.py
- 加入 Self-Attention
- 加入 BiGRU
- 加入 Bert
- ...

SentimentAnalysis

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README