mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 03:31:05 +06:00
Adding doctest for feature-extraction
. (#20240)
* Adding doctest for `feature-extraction`. * Update feature_extraction.py
This commit is contained in:
parent
529037fda5
commit
e434627858
@ -9,6 +9,19 @@ class FeatureExtractionPipeline(Pipeline):
|
||||
Feature extraction pipeline using no model head. This pipeline extracts the hidden states from the base
|
||||
transformer, which can be used as features in downstream tasks.
|
||||
|
||||
Example:
|
||||
|
||||
```python
|
||||
>>> from transformers import pipeline
|
||||
|
||||
>>> extractor = pipeline(model="bert-base-uncased", task="feature-extraction")
|
||||
>>> result = extractor("This is a simple test.", return_tensors=True)
|
||||
>>> result.shape # This is a tensor of shape [1, sequence_lenth, hidden_dimension] representing the input string.
|
||||
torch.Size([1, 8, 768])
|
||||
```
|
||||
|
||||
[Using pipelines in a webserver or with a dataset](../pipeline_tutorial)
|
||||
|
||||
This feature extraction pipeline can currently be loaded from [`pipeline`] using the task identifier:
|
||||
`"feature-extraction"`.
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user