Happy-Chen-CH/text_classification
This is a text classification project based on the "Category News" feature of Toutiao. Initially, we used the random forest algorithm, then upgraded to FastText, and finally built a new text classification model using a BERT pre-trained model with an added fully connected layer.There may still contain several bugs—let's work together to fix them.
GitHub repository with 38 stars and 1 forks.
Language: Python
Topics: ai, bert, distilbert, fasttext-python, python, random-forest-classifier, tensorflow, text-classification