transformers tf distilbert text-classification ditilbert text classification clinical notes wellnation en autotrain_compatible endpoints_compatible region:us
---
language:
- en
tags:
- ditilbert
- text classification
- clinical notes
- wellnation
---

<pre>
labels map = 
  {
  "0": "Gastroenterology",
  "1": "Neurology",
  "2": "Orthopedic",
  "3": "Radiology",
  "4": "Urology"
}
</pre>

<h2><i>The fine tuned model has been trained on around 2300 medical transcriptions, to classify medical specialty.
  More classes will be added as data would be available.</i></h2>