saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[pipeline] Alias NerPipeline as TokenClassificationPipeline

Browse source
This commit is contained in:
Julien Chaumond 2020-02-13 20:14:43 -05:00
parent 61a2b7dc9d
commit 7d22fefd37
2 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/transformers/__init__.py
View file

@ -101,6 +101,7 @@ from .pipelines import (
PipelineDataFormat,
QuestionAnsweringPipeline,
TextClassificationPipeline,
TokenClassificationPipeline,
pipeline,
)
from .tokenization_albert import AlbertTokenizer

3
src/transformers/pipelines.py
View file

@ -649,6 +649,9 @@ class NerPipeline(Pipeline):
return answers
TokenClassificationPipeline = NerPipeline
class QuestionAnsweringArgumentHandler(ArgumentHandler):
"""
QuestionAnsweringPipeline requires the user to provide multiple arguments (i.e. question & context) to be mapped