File tree 1 file changed +7
-4
lines changed
1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change 1
1
### Project: Classify Kaggle Consumer Finance Complaints
2
2
3
- ### Summary :
3
+ ### Highlights :
4
4
5
5
- This is a ** multi-class text classification** problem.
6
6
- The purpose of this project is to classify Kaggle Consumer Finance Complaints into ** 11 classes** .
18
18
19
19
### Train:
20
20
21
- - Command: ``` python3 train.py training_data.file training_parameters.file ```
21
+ - Command: python3 train.py training_data.file training_parameters.file
22
22
- 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 ``` .
23
25
24
26
### Predict:
25
27
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 ```
28
31
29
32
### Reference:
30
33
- [ Implement a cnn for text classification in tensorflow] ( http://www.wildml.com/2015/12/implementing-a-cnn-for-text-classification-in-tensorflow/ )
You can’t perform that action at this time.
0 commit comments