| from transformers import PretrainedConfig, AutoConfig | |
| from typing import List | |
| class BioNExtTaggerConfig(PretrainedConfig): | |
| model_type = "crf-tagger" | |
| def __init__( | |
| self, | |
| augmentation = "unk", | |
| context_size = 64, | |
| percentage_tags = 0.2, | |
| p_augmentation = 0.5, | |
| crf_reduction = "mean", | |
| version="0.1.2", | |
| **kwargs, | |
| ): | |
| self.version = version | |
| self.augmentation = augmentation | |
| self.context_size = context_size | |
| self.percentage_tags = percentage_tags | |
| self.p_augmentation = p_augmentation | |
| self.crf_reduction = crf_reduction | |
| super().__init__(**kwargs) | |