Docs: enhance README with Quickstart, examples, weights roadmap
#4
by
SHA888
- opened
README.md
CHANGED
|
@@ -23,7 +23,13 @@ use these as starting points for training or inference in medical NLP tasks.
|
|
| 23 |
- Classification config example (`examples/classification/`)
|
| 24 |
- Generation config example (`examples/generation/`)
|
| 25 |
|
| 26 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
```python
|
| 29 |
from medvllm.medical.config.models.medical_config import MedicalModelConfig
|
|
@@ -34,6 +40,39 @@ print(cfg.task_type)
|
|
| 34 |
Or directly load a specific example folder if exported as a subfolder with
|
| 35 |
its own config files.
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
## Medical Disclaimer
|
| 38 |
|
| 39 |
This repository and associated configurations are provided for research and
|
|
|
|
| 23 |
- Classification config example (`examples/classification/`)
|
| 24 |
- Generation config example (`examples/generation/`)
|
| 25 |
|
| 26 |
+
## Install
|
| 27 |
+
|
| 28 |
+
```bash
|
| 29 |
+
pip install medvllm
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
## Quickstart (Python)
|
| 33 |
|
| 34 |
```python
|
| 35 |
from medvllm.medical.config.models.medical_config import MedicalModelConfig
|
|
|
|
| 40 |
Or directly load a specific example folder if exported as a subfolder with
|
| 41 |
its own config files.
|
| 42 |
|
| 43 |
+
## Examples
|
| 44 |
+
|
| 45 |
+
- NER: [`examples/ner/config.json`](./examples/ner/config.json) | [`examples/ner/config.yaml`](./examples/ner/config.yaml)
|
| 46 |
+
- Classification: [`examples/classification/config.json`](./examples/classification/config.json) | [`examples/classification/config.yaml`](./examples/classification/config.yaml)
|
| 47 |
+
- Generation: [`examples/generation/config.json`](./examples/generation/config.json) | [`examples/generation/config.yaml`](./examples/generation/config.yaml)
|
| 48 |
+
|
| 49 |
+
Use these as starting points and customize fields like `task_type`, `classification_labels`, `medical_entity_types`, and domain settings.
|
| 50 |
+
|
| 51 |
+
## Tasks supported
|
| 52 |
+
|
| 53 |
+
- Named Entity Recognition (NER)
|
| 54 |
+
- Text Classification
|
| 55 |
+
- Text Generation
|
| 56 |
+
|
| 57 |
+
All tasks share a unified configuration schema via `MedicalModelConfig`.
|
| 58 |
+
|
| 59 |
+
## Weights roadmap
|
| 60 |
+
|
| 61 |
+
This repo currently focuses on configs. Model weights/adapters will be added progressively:
|
| 62 |
+
|
| 63 |
+
- BioBERT/ClinicalBERT adapters
|
| 64 |
+
- Task-specific fine-tuned checkpoints (NER/Classification)
|
| 65 |
+
|
| 66 |
+
Follow the repo for updates or open a Discussion to request specific checkpoints.
|
| 67 |
+
|
| 68 |
+
## Debug and logging
|
| 69 |
+
|
| 70 |
+
By default, verbose config debug prints are silenced. To enable them for troubleshooting, set:
|
| 71 |
+
|
| 72 |
+
```bash
|
| 73 |
+
export MEDVLLM_CONFIG_DEBUG=1
|
| 74 |
+
```
|
| 75 |
+
|
| 76 |
## Medical Disclaimer
|
| 77 |
|
| 78 |
This repository and associated configurations are provided for research and
|