← all repositories

jiegzhan/multi-class-text-classification-cnn

A convolutional neural network with word embeddings trained on TensorFlow/Keras to classify consumer finance complaints into 11 categories.

426 stars Python ML FrameworksDomain Apps
multi-class-text-classification-cnn
Velocity · 7d
+0.1
★ / day
Trend
steady
star history

This repository implements a multi-class text classification model using a Convolutional Neural Network and word embeddings. The model is built with TensorFlow 2 and Keras, trained on the Kaggle Consumer Finance Complaints dataset to predict product categories from complaint narratives. The workflow includes training scripts that produce a best model checkpoint and prediction scripts for inference on new data.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.