Он поддерживает пользовательские объекты, см. этот раздел под названием «Обучение дополнительному типу объекта».
Например, чтобы добавить метку с именем MY_ANIMAL
, вы можете использовать данные обучения вот так:
TRAIN_DATA = [
(
"Horses are too tall and they pretend to care about your feelings",
{"entities": [(0, 6, MY_ANIMAL)]},
),
("Do they bite?", {"entities": []}),
(
"horses are too tall and they pretend to care about your feelings",
{"entities": [(0, 6, MY_ANIMAL)]},
),
]
И добавьте это либо в существующую модель NER в качестве дополнительного обучения, либо во вновь созданную трубу NER.
Однако предостережение: модель ML оптимизирована для распознавания именованные сущности, которые обычно называются существительными, такими как «Джон», «Лондон» или «The Times». Вы также можете попробовать обучить его более общим c вещам, таким как числа, но это может не сработать.