Linhz/ViMNer

1a

2��dfv�@sDdd
�Zdd�Z
dd�Zdd�Zdd	�Zgd3�
Zee�
Zee�

dS)cCs*g}t||
�D]\}}|�
||f�

q|S)
u[4    Chuyển đổi danh sách từ và dự đoán sang định dạng (word, label)5    )�zip�append)�words�predictions�	formatted�word�label�r�AE:\demo_datn\pythonProject1\Model\MultimodelNER\Ner_processing.py�format_predictions
s


r6c
CsDg}
|D]6\}}d
|vr0|
�|�
d
d�|f�

q|
�||f�

q|
S)uc7    Tách các từ có dấu gạch dưới thành các từ riêng biệt với cùng nhãn8    �
_�
 )r�replace)rrrrrrr	�process_predictions9s



rc
Cs�g}
g}d}|D]�\}}|�d
�
rN|r>|
�
d�|�
|f�

g}|�
|�

|}q|�d�
r�|r�|dd�|dd�kr�|�
|�

q|r�|
�
d�|�
|f�

g}d}|
�
||f�

q|r�|
�
d�|�
|f�

|
S)N�B-r�I-���10startswithr�join)r�combinedZtemp_entityZ11temp_labelrrrrr	�combine_entitiess(



12



13

&





rc
Cs:g}
|D],\}}|�d
�
r&|dd�}|
�
||f�

q|
S)Nrr)rr)�entitiesZmodified_entitiesrrrrr	�remove_B_prefix4s


14


rc
Cs�g}
g}d}|D]�\}}|�d
�
rz|dd�}|durF|}|�
|�

q�||krZ|�
|�

q�|
�
d�|�
|f�

|g
}|}q|r�|
�
d�|�
|f�

g}d}|
�
||f�

q|r�|
�
d�|�
|f�

|
S)Nrrrr)�
tokens_labelsrZcurrent_combinationZ
current_label�tokenrrrr	�combine_i_tags=s,



15












r)Z)uDân�
O)utrír)uChứcr)uvôr)uđịchr)z	Euro 2008�EVENT-SPORT�uđầyr)uthuyếtr)uphụcr�ucủar�uTây Ban Nha�LOCATION)utrênr)uđấtr)uÁor!)uvàr)u16Thụy Sĩ�PERSON)uđãr)umởr)�rar)ukỷr)unguyênr)Zvinhr)Zquanhrr)�La�ORGANIZATION)ZFuria�I-ORGANIZATION-SPORTS)ZRojar&��
,r�uvớir)ulốir)uchơir)Ztiquir)Ztakarr)ubiếnr)uảor��
.r)ZTrongr)uquár)ukhứrr'r )unổir)utiếngrr))ubiệtr)Zdanhr)ZVuar�uvòngr�uloạirr*)uHọr)u	thườngr)Zthir�uđấur)urấtr)utốtr�uởrr,r-)unhưngr)ulạir)uchưar)Zbaor)ugiờr)uchứngr)Zminhr)uđượcr)usứcr)umạnhrr/r,)Zchungr)ukếtr)ugiảirr.)ulớnrr*)uLầnr)Zduyr)unhấtr)uhọr)ulênr)ungôir)ulàrr/)ukỳr)ZEuror)Z1964rr*N)r17rrrrrZcombined_tokens_labels�printrrrr	�<module>
s