我通过UIE进行训练10个epoch,结过有很大问题,示例:
>>> my_ie.set_schema(["产品名称", "产品属性"])
>>> my_ie("进口煤继续保持增长,导致供大于求,煤炭价格下降")
[{'产品名称': [{'text': '导', 'start': 10, 'end': 11, 'probability': 0.435139******5543}, {'text': '进口', 'start': 0, 'end': 2, 'probability': 0.362484******8954}, {'text': '持', 'start': 6, 'end': 7, 'probability': 0.525006******0286}, {'text': '名', 'probability': 0.482804******10835}, {'text': '于', 'start': 14, 'end': 15, 'probability': 0.429311******7655}, {'text': '继', 'start': 3, 'end': 4, 'probability': 0.520009******2779}, {'text': '增', 'start': 7, 'end': 8, 'probability': 0.446360******4402}, {'text': '品', 'probability': 0.486279******4239}], '产品属性': [{'text': '导', 'start': 10, 'end': 11, 'probability': 0.443422******25467}, {'text': '进口', 'start': 0, 'end': 2, 'probability': 0.361637120529565}, {'text': '持', 'start': 6, 'end': 7, 'probability': 0.531962624153266}, {'text': '于', 'start': 14, 'end': 15, 'probability': 0.436592******9145}, {'text': '继', 'start': 3, 'end': 4, 'probability': 0.527447******9435}, {'text': '增', 'start': 7, 'end': 8, 'probability': 0.450433******8278}, {'text': '性', 'probability': 0.419511******1352}, {'text': '品', 'probability': 0.483104******4145}]}]