This repository maintains the code for identifying social support and social isolation with fine-grained categories (presence or absence of social network, emotional support, instrumental support, general and loneliness) using rule- and large language models (LLM)-based algorithms.

python rule_based_classification.py

python llm_fine_tune_all.py

python llm_sentence_classification_fine_tuned.py

We will update the GPU version of the LLM code soon.

Please cite the following paper if you find this code useful.

Patra, B. G., Lepow, L. A., Kumar, P. K. R. J., Vekaria, V., Sharma, M. M., Adekkanattu, P., Fennessy, B., Hynes, G., Landi, I., Sanchez-Ruiz, J. A., Ryu, E., Biernacka, J. M., Nadkarni, G. M., Talati, A., Weissman, M., Olfson, M., Mann, J. J., Charney, A. W., & Pathak, J. (2024). Extracting Social Support and Social Isolation Information from Clinical Psychiatry Notes: Comparing a Rule-based NLP System and a Large Language Model. arXiv preprint arXiv:2403.17199)

Please contact brajagopal[dot]cse[at]gmail[dot]com for further information.