Skip to content

Commit 614044e

Browse files
authored
Update README.md
1 parent 0616ceb commit 614044e

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

README.md

+7-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
### Project: Classify Kaggle Consumer Finance Complaints
22

3-
### Summary:
3+
### Highlights:
44

55
- This is a **multi-class text classification** problem.
66
- The purpose of this project is to classify Kaggle Consumer Finance Complaints into **11 classes**.
@@ -18,13 +18,16 @@
1818

1919
### Train:
2020

21-
- Command: ```python3 train.py training_data.file training_parameters.file```
21+
- Command: python3 train.py training_data.file training_parameters.file
2222
- Example: ```python3 train.py ./data/consumer_complaints.csv.zip ./parameters.json```
23+
24+
A directory will be generated during training, and it contains model which will be loaded when running ```predict.py```.
2325

2426
### Predict:
2527

26-
- Command: ```python3 predict.py ./trained_model_directory/ test_samples.file```
27-
- Example: ```python3 predict.py ./trained_model_1478649295/ ./data/small_samples.json```
28+
Provide the model directory (generated when running ```train.py```) and test file to ```predict.py```.
29+
- Command: python3 predict.py ./trained_model_directory/ test_samples.file
30+
- Example: ```python3 predict.py ./trained_model_1479757124/ ./data/small_samples.json```
2831

2932
### Reference:
3033
- [Implement a cnn for text classification in tensorflow](http://www.wildml.com/2015/12/implementing-a-cnn-for-text-classification-in-tensorflow/)

0 commit comments

Comments
 (0)