Sentence Level Text Classification with Convolutional Neural Networks
- This is a multi-class text classification (sentence classification) problem.
- The goal of this project is to classify Consumer Complaints into classes.
- Also Extract if some Maintenance issue is mentioned in customer reivew.
- The model was built with Convolutional Neural Network (CNN) and Word Embeddings on Tensorflow.
- Python 3
- Tensorflow
- Numpy
- Pandas
-
Input: consumer_complaint_narrative
-
Output: product
- Command: python3 train.py training_data.file parameters.json
- Example:
python3 train.py ./data/consumer_complaints.csv.zip ./parameters.json
A directory will be created during training, and the trained model will be saved in this directory.
Provide the model directory (created when running train.py
) and new data to predict.py
.
- Command: python3 predict.py ./trained_model_directory/ new_data.file
- Example:
python3 predict.py ./trained_model_1479757124/ ./data/small_samples.json