Showing
1 changed file
with
2 additions
and
2 deletions
| ... | @@ -8,8 +8,8 @@ The main method follows the next pipeline: | ... | @@ -8,8 +8,8 @@ The main method follows the next pipeline: |
| 8 | - Parse abstracts from two input files (classA and classB; see files format at the `data/` directory) | 8 | - Parse abstracts from two input files (classA and classB; see files format at the `data/` directory) |
| 9 | - Transform abstracts into their TFIDF sparse representations | 9 | - Transform abstracts into their TFIDF sparse representations |
| 10 | - Train Support Vector Machines with different parameters by using GridSearch | 10 | - Train Support Vector Machines with different parameters by using GridSearch |
| 11 | -- Select the best estimator and save it at `model/svm_model.pkl` (default) | 11 | +- Select the best estimator and save it at `model_binClass/svm_model.pkl` (default) |
| 12 | -- Save TFIDF transformation for keeping the training vocabulary (stored at `model/tfidf_model.pkl`) | 12 | +- Save TFIDF transformation for keeping the training vocabulary (stored at `model_binClass/tfidf_model.pkl`) |
| 13 | 13 | ||
| 14 | ### Prediction mode | 14 | ### Prediction mode |
| 15 | - Parse abstracts from a unique input file | 15 | - Parse abstracts from a unique input file | ... | ... |
-
Please register or login to post a comment