python for문 if condition(BIO tag deletion)

(코드 복붙용)

NER task에서 BIO tag를 삭제할 때

ex ) "B-key", "I-price", "I-price", "O-price" ==> B-price", "I-price", "I-price", "O-price"로 변경할 때

 

 

    from collections import Counter
    
    for idx, bbox in enumerate(split_lst):
        lst = [box[2:] if box[:2] in ["B-", "I-", "O-"] else pdb.set_trace() for box in bbox]
        counter = Counter(lst)
        max_class = max(counter, key= counter.get)
        split_lst[idx] = [box[:2]+max_class for box in bbox]