diff --git a/README.md b/README.md index 58df8bd..e157b64 100644 --- a/README.md +++ b/README.md @@ -6,18 +6,18 @@ - This model was built with **CNN, RNN (LSTM and GRU) and Word Embeddings** on **Tensorflow**. ### Data: [Kaggle San Francisco Crime](https://www.kaggle.com/c/sf-crime/data) - - Input: **Descript** + - Input: **Description** - Output: **Category** - Examples: - Descript | Category + Description | 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 - + ### Train: - Command: python3 train.py train_data.file train_parameters.json - Example: ```python3 train.py ./data/train.csv.zip ./training_config.json``` @@ -25,6 +25,6 @@ ### Predict: - Command: python3 predict.py ./trained_results_dir/ new_data.csv - Example: ```python3 predict.py ./trained_results_1478563595/ ./data/small_samples.csv``` - + ### Reference: - [Implement a cnn for text classification in tensorflow](http://www.wildml.com/2015/12/implementing-a-cnn-for-text-classification-in-tensorflow/)