| license: mit | |
| language: | |
| - en | |
| library_name: transformers | |
| pipeline_tag: text-classification | |
| tags: | |
| - bert-base-uncased | |
| - disease | |
| - medical | |
| A model that classifies a text like "I am feeling vomiting, breathlessness, and sweating" into it's disease (2 - 'Heart attack'). | |
| Basically a symptoms to disease classification. It's a baseline for a chatbot I am working on that will classify user's symptoms into diseases in a natural language environment. |