
<!-- This model card has been generated automatically according to the information Keras had access to. You should
probably proofread and complete it, then remove this comment. -->

## Model description

This model is trained on a Symptoms to Disease [Dataset]("https://www.kaggle.com/datasets/niyarrbarman/symptom2disease"). 
The model takes in a symptom prompt and classifies the most probable disease from it. 


### Training results

| Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
|:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
| 2.6613     | 0.2954         | 1.7206          | 0.7167              | 0     |
| 1.1325     | 0.8491         | 0.6683          | 0.9000              | 1     |
| 0.4055     | 0.9750         | 0.2890          | 0.9417              | 2     |
| 0.1551     | 0.9972         | 0.1912          | 0.9667              | 3     |
| 0.0803     | 1.0            | 0.1296          | 0.9583              | 4     |


### Framework versions

- Transformers 4.28.1
- TensorFlow 2.12.0
- Tokenizers 0.13.3
{"language": ["en"], "license": "apache-2.0", "tags": ["generated_from_keras_callback"], "model-index": [{"name": "abhirajeshbhai/symptom-2-disease-net", "results": []}]}
transformers tf bert text-classification generated_from_keras_callback en license:apache-2.0 autotrain_compatible endpoints_compatible has_space region:us
TAGS
#transformers #tf #bert #text-classification #generated_from_keras_callback #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #has_space #region-us 

Model description
-----------------


This model is trained on a Symptoms to Disease Dataset.
The model takes in a symptom prompt and classifies the most probable disease from it.


### Training results



### Framework versions


* Transformers 4.28.1
* TensorFlow 2.12.0
* Tokenizers 0.13.3



### Training results ### Framework versions


* Transformers 4.28.1
* TensorFlow 2.12.0
* Tokenizers 0.13.3
TAGS
#transformers #tf #bert #text-classification #generated_from_keras_callback #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #has_space #region-us 
 ### Training results ### Framework versions


* Transformers 4.28.1
* TensorFlow 2.12.0
* Tokenizers 0.13.3
TAGS
#transformers #tf #bert #text-classification #generated_from_keras_callback #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #has_space #region-us 
### Training results### Framework versions


* Transformers 4.28.1
* TensorFlow 2.12.0
* Tokenizers 0.13.3