Classify Kaggle San Francisco Crime Description into 39 classes. Build the model with CNN, RNN (GRU and LSTM) and Word Embeddings on Tensorflow.
Input: Descript
Output: Category
Examples:
Descript | Category |
---|---|
GRAND THEFT FROM LOCKED AUTO | LARCENY/THEFT |
POSSESSION OF NARCOTICS PARAPHERNALIA | DRUG/NARCOTIC |
AIDED CASE, MENTAL DISTURBED | NON-CRIMINAL |
AGGRAVATED ASSAULT WITH BODILY FORCE | ASSAULT |
ATTEMPTED ROBBERY ON THE STREET WITH A GUN | ROBBERY |
python3 train.py ./data/train.csv.zip ./training_config.json
python3 predict.py ./trained_results_1478563595/ ./data/small_samples.csv