Showing
1 changed file
with
1 additions
and
0 deletions
| ... | @@ -7,6 +7,7 @@ The main method follows the next pipeline: | ... | @@ -7,6 +7,7 @@ The main method follows the next pipeline: |
| 7 | ### Training mode | 7 | ### Training mode |
| 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 | +- Transform TFIDF representations into their 200-dimensional SVD approximation and save it at `model_binClass/svd_model.pkl` | ||
| 10 | - Train Support Vector Machines with different parameters by using GridSearch | 11 | - Train Support Vector Machines with different parameters by using GridSearch |
| 11 | - Select the best estimator and save it at `model_binClass/svm_model.pkl` (default) | 12 | - 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_binClass/tfidf_model.pkl`) | 13 | - Save TFIDF transformation for keeping the training vocabulary (stored at `model_binClass/tfidf_model.pkl`) | ... | ... |
-
Please register or login to post a comment