SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix typo in FeatureExtractionPipeline docstring

Browse Source
This commit is contained in:
Lorenzo Ampil
2020-04-03 14:58:53 +08:00
committed by Julien Chaumond
parent 715aa5b135
commit 747907dc5e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines.py
View File

@@ -495,7 +495,7 @@ class Pipeline(_ScikitCompat):
class FeatureExtractionPipeline(Pipeline): class FeatureExtractionPipeline(Pipeline):
""" """
Feature extraction pipeline using Model head. This pipeline extracts the hidden states from the base transformer, Feature extraction pipeline using Model head. This pipeline extracts the hidden states from the base transformer,
which can be used as features in a downstream tasks. which can be used as features in downstream tasks.
This feature extraction pipeline can currently be loaded from the :func:`~transformers.pipeline` method using This feature extraction pipeline can currently be loaded from the :func:`~transformers.pipeline` method using
the following task identifier(s): the following task identifier(s):