shaadclt / Training-SmallLanguageModel-DiseaseSymptoms

This project utilizes the power of language models, specifically the GPT-2 model, to predict medical symptoms based on input text. By fine-tuning the GPT-2 model on a dataset containing disease names and associated symptoms, we train a language model to generate probable symptoms for a given disease.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

shaadclt/Training-SmallLanguageModel-DiseaseSymptoms Stargazers