This program utilizes TF-IDF and SVM to predict patient engagement in text-based suicide hotline.
To run the program, execute all the code in driver.py
file under python
environment. The program will run the pipeline
and report the result of model evaluation. To tune the model, hyperparameters are set in driver.py
file.
The packages that may require installation: spaCy, sklearn, numpy, matplotlib, pandas
To intall spaCy
, follow the link https://spacy.io/usage. To install sklearn
, follow the link
https://scikit-learn.org/stable/install.html. Other packages can be installed directly
through pip
or conda
.