This repository contains Keras implementations for Character-level Convolutional Neural Networks for text classification on AG's News Topic Classification Dataset.
The following models have been implemented:
Kim's CharCNN was originally part of an end-to-end trained pipeline for language modelling, but has been adapted for text classification.
$ pip install -r requirements.txt
Specify the training and testing data sources and model hyperparameters in the config.json
file.
Run the main.py file as below:
$ python main.py --model [model_name]
Replace [model_name]
with either zhang
or kim
to run the desired model.
Coming soon.