akirtn / lab_tips_code Goto Github PK

def write_file(out_file, docs):  
    """データセットをファイルに書き出す"""  
    with open(out_file, "w") as f:  
        for doc in docs:  
            for result in doc:  
                for mrph in result.mrph_list():  
                    match = re.search(r"<NE:(.+?):([BI])/>", mrph.fstring)  
                    if match:  
                        # B-PERSONのような形式の固有表現ラベルを作成  
                        ne_tag = "{}-{}".format(match.group(2),match.group(1))
                    else:  
                        # 固有表現ラベルの無い場合は"O"ラベルを付与  
                        ne_tag = "O"  
                    # 1カラム目に単語、4カラム目に固有表現ラベルを書く  
                    # それ以外のカラムは利用しない  
                    f.write("{} N/A N/A {}\n".format(mrph.midasi, ne_tag))  
                f.write("\n")