File size: 1,405 Bytes
9057844 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
license: apache-2.0
language:
- fr
- en
base_model:
- google-bert/bert-base-uncased
pipeline_tag: text-classification
library_name: sentence-transformers
---
# Takeda Section Classifier
Pretrained model (finetuned version of [BERT Multilingual Uncased](https://huggingface.co/google-bert/bert-base-multilingual-uncased)) on french and english documents using supervised training for sections classification.
This work has been made by Digital Innovation Team from Belgium 🇧🇪 (LE).
## Model Description
The model aims at classifying text in classes representing part of reports:
* Description
* Immediate Correction
* Root Cause
* Action Plan
* Impacted Elements
## Intended uses & limitations
The model can be use for Takeda documentation, the team do not guarantee results for out of the scope documentation.
## How to Use
You can use this model directly with a pipeline for text classification:
```python
from transformers import (
TextClassificationPipeline,
AutoTokenizer,
AutoModelForSequenceClassification,
)
tokenizer = AutoTokenizer.from_pretrained("TakedaAIML/section_classifier")
model = AutoModelForSequenceClassification.from_pretrained(
"TakedaAIML/section_classifier"
)
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
prediction = pipe('this is a piece of text representing the Description section. An event occur on june 24 and ...')
``` |