# keras_lstm_chinese_document_classification

**Repository Path**: tunmx/keras_lstm_chinese_document_classification

## Basic Information

- **Project Name**: keras_lstm_chinese_document_classification
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2020-07-21
- **Last Updated**: 2020-12-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 下载代码
git clone https://github.com/Jinkeycode/keras_lstm_chinese_document_classification.git

# 安装依赖
pip install -r requirements.txt

# 二分类
标签形式为[0, 0, 1, 1] 代表分类的四个样本分别为第 0 类和第 1 类

具体代码在 `chinese_binary.py`

# 多分类
标签形式为[[1, 0, 0], [0, 1, 0], [0, 0, 1]] 代表分类的四个样本分别为第 0 类、第 1 类、第 2 类

具体代码在 `chinese_multi.py`

# 训练
取消掉这几行的注释即可
```
# 训练模型
# model.fit(text_processed, target, batch_size=512, epochs=10, )
# 保存模型
# model.save("health_and_tech.h5")
```

# 预测
sen = "*" 改为 自己的句子