o qi@s2dZddlZddlmZdededefddZdS) zNER Tensorflow ModelN)CRFn_wordsn_tagsmax_lenc Cstjjj|fdd}tjjj|ddddd|}tjjtjjjddd d d |}tjjd |}tjj||}t |}||}tjj ||} | S) z Ner model input_layer)shapename2Tembedding_layer) input_dim output_dim mask_zerorrelug?)unitsreturn_sequences activationrecurrent_dropout) tfkeraslayersInput Embedding BidirectionalLSTMDropoutDensermodelsModel) rrrinput_r lstm_layertensorcrfoutputmodelr&M/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/models/ner/model.py_bi_lstm_crf_models$r()__doc__ tensorflowrtf2crfrintr(r&r&r&r's