https://github.com/chmoder/multi-class-text-classification-trainer
https://github.com/chmoder/multi-class-text-classification-trainer
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/chmoder/multi-class-text-classification-trainer
- Owner: chmoder
- License: mit
- Created: 2020-06-22T01:51:30.000Z (about 5 years ago)
- Default Branch: master
- Last Pushed: 2024-09-01T20:30:56.000Z (10 months ago)
- Last Synced: 2025-02-16T05:14:33.170Z (4 months ago)
- Language: Python
- Size: 11.7 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Multi-Class Text Classification Trainer (TensorFlow 2)
The goal for this project is to make an easy to use multi-class text classifier;
More like an application than a data science project.
All you need to do is format your data in a CSV like so:
```
Sentences of text,label1;label2
```
The labels are separated by a semicolon. You don't need multiple, but does can support multiple.
#### Setup:
1) Replace data.csv with your data.
2) Set up a virtual environment, and install the requirements
3) Run main.py
The requirement.txt file in this project assumes we want to use `tensorflow-gpu`.
Feel free to review the requirements and adjust versions, etc.