multitask-classifying-emotion-toxicity-valorant-chat
/
pipeline
/model
/model_with_bert_multilingual.py
from transformers import TFBertModel | |
import tensorflow as tf | |
from tensorflow.keras.layers import Input, Dense, LSTM, Bidirectional # type: ignore | |
from tensorflow.keras.models import Model # type: ignore | |
from tensorflow.keras.optimizers import Adam # type: ignore | |
from tensorflow.keras.regularizers import l1_l2 # type: ignore | |
def build_model_multilingual(max_length = 65, layer = 40,dropout = 0.69, l2_lstm = 0.01, learning_rate = 1e-4): | |