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