[735bb5]: / src / ml_models / bert / config.py

Download this file

15 lines (12 with data), 410 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
# 3rd-Party Dependencies
# ----------------------
from transformers import AutoConfig
# Constants
# ---------
from constants import MODELS, MODELS_CACHE_DIR
ClinicalBERTConfig = AutoConfig.from_pretrained(
pretrained_model_name_or_path=MODELS["bert"]["clinical-bert"],
cache_dir=MODELS_CACHE_DIR,
)
ClinicalBERTConfig.hidden_dropout_prob = 0.25
ClinicalBERTConfig.attention_probs_dropout_prob = 0.25