https://github.com/chmoder/multi-class-text-classification-trainer

https://github.com/chmoder/multi-class-text-classification-trainer

Last synced: 13 days 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 (over 4 years ago)
• Default Branch: master
• Last Pushed: 2023-03-24T22:50:43.000Z (over 1 year ago)
• Last Synced: 2023-04-16T12:07:42.179Z (over 1 year ago)
• Language: Python
• Size: 8.79 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.