16 lines
542 B
Python
16 lines
542 B
Python
|
from lightkg.erl import NER
|
||
|
|
|
||
|
|
ner_model = NER()
|
||
|
|
|
||
|
|
train_path = '/home/lightsmile/NLP/corpus/ner/train.sample.txt'
|
||
|
|
dev_path = '/home/lightsmile/NLP/corpus/ner/test.sample.txt'
|
||
|
|
vec_path = '/home/lightsmile/NLP/embedding/char/token_vec_300.bin'
|
||
|
|
|
||
|
|
# ner_model.train(train_path, vectors_path=vec_path, dev_path=dev_path, save_path='./ner_saves')
|
||
|
|
|
||
|
|
ner_model.load('./ner_saves')
|
||
|
|
# ner_model.test(train_path)
|
||
|
|
|
||
|
|
from pprint import pprint
|
||
|
|
pprint(ner_model.predict('另一个很酷的事情是，通过框架我们可以停止并在稍后恢复训练。'))
|