huggingface--transformers
e06fe8e8c6
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
New model PR merged notification / Notify new model (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
436 ่ก
24 KiB
Markdown
436 ่ก
24 KiB
Markdown
<!--Copyright 2022 The HuggingFace Team. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
the License. You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
specific language governing permissions and limitations under the License.
|
|
|
|
โ ๏ธ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
|
|
rendered properly in your Markdown viewer.
|
|
|
|
-->
|
|
|
|
# ๋๋ฌ๋ณด๊ธฐ [[quick-tour]]
|
|
|
|
[[open-in-colab]]
|
|
|
|
๐ค Transformers๋ฅผ ์์ํด๋ณด์ธ์! ๊ฐ๋ฐํด๋ณธ ์ ์ด ์๋๋ผ๋ ์ฝ๊ฒ ์ฝ์ ์ ์๋๋ก ์ฐ์ธ ์ด ๊ธ์ [`pipeline`](./main_classes/pipelines)์ ์ฌ์ฉํ์ฌ ์ถ๋ก ํ๊ณ , ์ฌ์ ํ์ต๋ ๋ชจ๋ธ๊ณผ ์ ์ฒ๋ฆฌ๊ธฐ๋ฅผ [AutoClass](./model_doc/auto)๋ก ๋ก๋ํ๊ณ , PyTorch ๋๋ TensorFlow๋ก ๋ชจ๋ธ์ ๋น ๋ฅด๊ฒ ํ์ต์ํค๋ ๋ฐฉ๋ฒ์ ์๊ฐํด ๋๋ฆด ๊ฒ์
๋๋ค. ๋ณธ ๊ฐ์ด๋์์ ์๊ฐ๋๋ ๊ฐ๋
์ (ํนํ ์ด๋ณด์์ ๊ด์ ์ผ๋ก) ๋ ์น์ ํ๊ฒ ์ ํ๊ณ ์ถ๋ค๋ฉด, ํํ ๋ฆฌ์ผ์ด๋ [์ฝ์ค](https://huggingface.co/course/chapter1/1)๋ฅผ ์ฐธ์กฐํ๊ธฐ๋ฅผ ๊ถ์ฅํฉ๋๋ค.
|
|
|
|
์์ํ๊ธฐ ์ ์ ํ์ํ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๊ฐ ๋ชจ๋ ์ค์น๋์ด ์๋์ง ํ์ธํ์ธ์:
|
|
|
|
```bash
|
|
!pip install transformers datasets evaluate accelerate
|
|
```
|
|
|
|
๋ํ ์ ํธํ๋ ๋จธ์ ๋ฌ๋ ํ๋ ์์ํฌ๋ฅผ ์ค์นํด์ผ ํฉ๋๋ค:
|
|
|
|
|
|
```bash
|
|
pip install torch
|
|
```
|
|
|
|
## ํ์ดํ๋ผ์ธ [[pipeline]]
|
|
|
|
<Youtube id="tiZFewofSLM"/>
|
|
|
|
[`pipeline`](./main_classes/pipelines)์ ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ๋ก ์ถ๋ก ํ๊ธฐ์ ๊ฐ์ฅ ์ฝ๊ณ ๋น ๋ฅธ ๋ฐฉ๋ฒ์
๋๋ค. [`pipeline`]์ ์ฌ๋ฌ ๋ชจ๋ฌ๋ฆฌํฐ์์ ๋ค์ํ ๊ณผ์
์ ์ฝ๊ฒ ์ฒ๋ฆฌํ ์ ์์ผ๋ฉฐ, ์๋ ํ์ ํ์๋ ๋ช ๊ฐ์ง ๊ณผ์
์ ๊ธฐ๋ณธ์ ์ผ๋ก ์ง์ํฉ๋๋ค:
|
|
|
|
<Tip>
|
|
|
|
์ฌ์ฉ ๊ฐ๋ฅํ ์์
์ ์ ์ฒด ๋ชฉ๋ก์ [Pipelines API ์ฐธ์กฐ](./main_classes/pipelines)๋ฅผ ํ์ธํ์ธ์.
|
|
|
|
</Tip>
|
|
|
|
| **ํ์คํฌ** | **์ค๋ช
** | **๋ชจ๋ฌ๋ฆฌํฐ** | **ํ์ดํ๋ผ์ธ ID** |
|
|
|-----------------|----------------------------------------------------------------------|------------------|-----------------------------------------------|
|
|
| ํ
์คํธ ๋ถ๋ฅ | ํ
์คํธ์ ์๋ง์ ๋ ์ด๋ธ ๋ถ์ด๊ธฐ | ์์ฐ์ด ์ฒ๋ฆฌ(NLP) | pipeline(task="sentiment-analysis") |
|
|
| ํ
์คํธ ์์ฑ | ์ฃผ์ด์ง ๋ฌธ์์ด ์
๋ ฅ๊ณผ ์ด์ด์ง๋ ํ
์คํธ ์์ฑํ๊ธฐ | ์์ฐ์ด ์ฒ๋ฆฌ(NLP) | pipeline(task="text-generation") |
|
|
| ๊ฐ์ฒด๋ช
์ธ์ | ๋ฌธ์์ด์ ๊ฐ ํ ํฐ๋ง๋ค ์๋ง์ ๋ ์ด๋ธ ๋ถ์ด๊ธฐ (์ธ๋ฌผ, ์กฐ์ง, ์ฅ์ ๋ฑ๋ฑ) | ์์ฐ์ด ์ฒ๋ฆฌ(NLP) | pipeline(task="ner") |
|
|
| ์ง์์๋ต | ์ฃผ์ด์ง ๋ฌธ๋งฅ๊ณผ ์ง๋ฌธ์ ๋ฐ๋ผ ์ฌ๋ฐ๋ฅธ ๋๋ตํ๊ธฐ | ์์ฐ์ด ์ฒ๋ฆฌ(NLP) | pipeline(task="question-answering") |
|
|
| ๋น์นธ ์ฑ์ฐ๊ธฐ | ๋ฌธ์์ด์ ๋น์นธ์ ์๋ง์ ํ ํฐ ๋ง์ถ๊ธฐ | ์์ฐ์ด ์ฒ๋ฆฌ(NLP) | pipeline(task="fill-mask") |
|
|
| ์ด๋ฏธ์ง ๋ถ๋ฅ | ์ด๋ฏธ์ง์ ์๋ง์ ๋ ์ด๋ธ ๋ถ์ด๊ธฐ | ์ปดํจํฐ ๋น์ (CV) | pipeline(task="image-classification") |
|
|
| ์ด๋ฏธ์ง ๋ถํ | ์ด๋ฏธ์ง์ ํฝ์
๋ง๋ค ๋ ์ด๋ธ ๋ถ์ด๊ธฐ(์๋งจํฑ, ํ๋ํฑ ๋ฐ ์ธ์คํด์ค ๋ถํ ํฌํจ) | ์ปดํจํฐ ๋น์ (CV) | pipeline(task="image-segmentation") |
|
|
| ๊ฐ์ฒด ํ์ง | ์ด๋ฏธ์ง ์ ๊ฐ์ฒด์ ๊ฒฝ๊ณ ์์๋ฅผ ๊ทธ๋ฆฌ๊ณ ํด๋์ค๋ฅผ ์์ธกํ๊ธฐ | ์ปดํจํฐ ๋น์ (CV) | pipeline(task="object-detection") |
|
|
| ์ค๋์ค ๋ถ๋ฅ | ์ค๋์ค ํ์ผ์ ์๋ง์ ๋ ์ด๋ธ ๋ถ์ด๊ธฐ | ์ค๋์ค | pipeline(task="audio-classification") |
|
|
| ์๋ ์์ฑ ์ธ์ | ์ค๋์ค ํ์ผ ์ ์์ฑ์ ํ
์คํธ๋ก ๋ฐ๊พธ๊ธฐ | ์ค๋์ค | pipeline(task="automatic-speech-recognition") |
|
|
| ์๊ฐ ์ง์์๋ต | ์ฃผ์ด์ง ์ด๋ฏธ์ง์ ์ง๋ฌธ์ ๋ํด ์ฌ๋ฐ๋ฅด๊ฒ ๋๋ตํ๊ธฐ | ๋ฉํฐ๋ชจ๋ฌ | pipeline(task="vqa") |
|
|
| ๋ฌธ์ ์ง์์๋ต | ์ฃผ์ด์ง ๋ฌธ์์ ์ง๋ฌธ์ ๋ํด ์ฌ๋ฐ๋ฅด๊ฒ ๋๋ตํ๊ธฐ | ๋ฉํฐ๋ชจ๋ฌ | pipeline(task="document-question-answering") |
|
|
| ์ด๋ฏธ์ง ์บก์
๋ฌ๊ธฐ | ์ฃผ์ด์ง ์ด๋ฏธ์ง์ ์บก์
์์ฑํ๊ธฐ | ๋ฉํฐ๋ชจ๋ฌ | pipeline(task="image-to-text") |
|
|
|
|
๋จผ์ [`pipeline`]์ ์ธ์คํด์ค๋ฅผ ์์ฑํ๊ณ ์ฌ์ฉํ ์์
์ ์ง์ ํฉ๋๋ค. ์ด ๊ฐ์ด๋์์๋ ๊ฐ์ ๋ถ์์ ์ํด [`pipeline`]์ ์ฌ์ฉํ๋ ์์ ๋ฅผ ๋ณด์ฌ๋๋ฆฌ๊ฒ ์ต๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import pipeline
|
|
|
|
>>> classifier = pipeline("sentiment-analysis")
|
|
```
|
|
|
|
[`pipeline`]์ ๊ฐ์ ๋ถ์์ ์ํ [์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ](https://huggingface.co/distilbert/distilbert-base-uncased-finetuned-sst-2-english)๊ณผ ํ ํฌ๋์ด์ ๋ฅผ ์๋์ผ๋ก ๋ค์ด๋ก๋ํ๊ณ ์บ์ํฉ๋๋ค. ์ด์ `classifier`๋ฅผ ๋์ ํ
์คํธ์ ์ฌ์ฉํ ์ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> classifier("We are very happy to show you the ๐ค Transformers library.")
|
|
[{'label': 'POSITIVE', 'score': 0.9998}]
|
|
```
|
|
|
|
๋ง์ฝ ์
๋ ฅ์ด ์ฌ๋ฌ ๊ฐ ์๋ ๊ฒฝ์ฐ, ์
๋ ฅ์ ๋ฆฌ์คํธ๋ก [`pipeline`]์ ์ ๋ฌํ์ฌ, ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ์ ์ถ๋ ฅ์ ๋์
๋๋ฆฌ๋ก ์ด๋ฃจ์ด์ง ๋ฆฌ์คํธ ํํ๋ก ๋ฐ์ ์ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> results = classifier(["We are very happy to show you the ๐ค Transformers library.", "We hope you don't hate it."])
|
|
>>> for result in results:
|
|
... print(f"label: {result['label']}, with score: {round(result['score'], 4)}")
|
|
label: POSITIVE, with score: 0.9998
|
|
label: NEGATIVE, with score: 0.5309
|
|
```
|
|
|
|
[`pipeline`]์ ์ฃผ์ด์ง ๊ณผ์
์ ๊ด๊ณ์์ด ๋ฐ์ดํฐ์
์ ๋ถ๋ฅผ ์ํํ ์๋ ์์ต๋๋ค. ์ด ์์ ์์๋ ์๋ ์์ฑ ์ธ์์ ๊ณผ์
์ผ๋ก ์ ํํด ๋ณด๊ฒ ์ต๋๋ค:
|
|
|
|
```py
|
|
>>> import torch
|
|
>>> from transformers import pipeline
|
|
|
|
>>> speech_recognizer = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")
|
|
```
|
|
|
|
๋ฐ์ดํฐ์
์ ๋ก๋ํ ์ฐจ๋ก์
๋๋ค. (์์ธํ ๋ด์ฉ์ ๐ค Datasets [์์ํ๊ธฐ](https://huggingface.co/docs/datasets/quickstart#audio)์ ์ฐธ์กฐํ์ธ์) ์ฌ๊ธฐ์์๋ [MInDS-14](https://huggingface.co/datasets/PolyAI/minds14) ๋ฐ์ดํฐ์
์ ๋ก๋ํ๊ฒ ์ต๋๋ค:
|
|
|
|
```py
|
|
>>> from datasets import load_dataset, Audio
|
|
|
|
>>> dataset = load_dataset("PolyAI/minds14", name="en-US", split="train") # doctest: +IGNORE_RESULT
|
|
```
|
|
|
|
๋ฐ์ดํฐ์
์ ์ํ๋ง ๋ ์ดํธ๊ฐ ๊ธฐ์กด ๋ชจ๋ธ์ธ [`facebook/wav2vec2-base-960h`](https://huggingface.co/facebook/wav2vec2-base-960h)์ ํ๋ จ ๋น์ ์ํ๋ง ๋ ์ดํธ์ ์ผ์นํ๋์ง ํ์ธํด์ผ ํฉ๋๋ค:
|
|
|
|
```py
|
|
>>> dataset = dataset.cast_column("audio", Audio(sampling_rate=speech_recognizer.feature_extractor.sampling_rate))
|
|
```
|
|
|
|
`"audio"` ์ด์ ํธ์ถํ๋ฉด ์๋์ผ๋ก ์ค๋์ค ํ์ผ์ ๊ฐ์ ธ์์ ๋ฆฌ์ํ๋งํฉ๋๋ค. ์ฒซ 4๊ฐ ์ํ์์ ์์ ์จ์ด๋ธํผ ๋ฐฐ์ด์ ์ถ์ถํ๊ณ ํ์ดํ๋ผ์ธ์ ๋ฆฌ์คํธ๋ก ์ ๋ฌํ์ธ์:
|
|
|
|
```py
|
|
>>> result = speech_recognizer(dataset[:4]["audio"])
|
|
>>> print([d["text"] for d in result])
|
|
['I WOULD LIKE TO SET UP A JOINT ACCOUNT WITH MY PARTNER HOW DO I PROCEED WITH DOING THAT', "FONDERING HOW I'D SET UP A JOIN TO HELL T WITH MY WIFE AND WHERE THE AP MIGHT BE", "I I'D LIKE TOY SET UP A JOINT ACCOUNT WITH MY PARTNER I'M NOT SEEING THE OPTION TO DO IT ON THE APSO I CALLED IN TO GET SOME HELP CAN I JUST DO IT OVER THE PHONE WITH YOU AND GIVE YOU THE INFORMATION OR SHOULD I DO IT IN THE AP AN I'M MISSING SOMETHING UQUETTE HAD PREFERRED TO JUST DO IT OVER THE PHONE OF POSSIBLE THINGS", 'HOW DO I FURN A JOINA COUT']
|
|
```
|
|
|
|
์์ฑ์ด๋ ๋น์ ๊ณผ ๊ฐ์ด ์
๋ ฅ์ด ํฐ ๋๊ท๋ชจ ๋ฐ์ดํฐ์
์ ๊ฒฝ์ฐ, ๋ชจ๋ ์
๋ ฅ์ ๋ฉ๋ชจ๋ฆฌ์ ๋ก๋ํ๋ ค๋ฉด ๋ฆฌ์คํธ ๋์ ์ ๋๋ ์ดํฐ ํํ๋ก ์ ๋ฌํด์ผ ํฉ๋๋ค. ์์ธํ ๋ด์ฉ์ [Pipelines API ์ฐธ์กฐ](./main_classes/pipelines)๋ฅผ ํ์ธํ์ธ์.
|
|
|
|
### ํ์ดํ๋ผ์ธ์์ ๋ค๋ฅธ ๋ชจ๋ธ๊ณผ ํ ํฌ๋์ด์ ์ฌ์ฉํ๊ธฐ [[use-another-model-and-tokenizer-in-the-pipeline]]
|
|
|
|
[`pipeline`]์ [Hub](https://huggingface.co/models)์ ๋ชจ๋ ๋ชจ๋ธ์ ์ฌ์ฉํ ์ ์๊ธฐ ๋๋ฌธ์, [`pipeline`]์ ๋ค๋ฅธ ์ฉ๋์ ๋ง๊ฒ ์ฝ๊ฒ ์์ ํ ์ ์์ต๋๋ค. ์๋ฅผ ๋ค์ด, ํ๋์ค์ด ํ
์คํธ๋ฅผ ์ฒ๋ฆฌํ ์ ์๋ ๋ชจ๋ธ์ ์ฌ์ฉํ๊ธฐ ์ํด์ Hub์ ํ๊ทธ๋ฅผ ์ฌ์ฉํ์ฌ ์ ์ ํ ๋ชจ๋ธ์ ํํฐ๋งํ๋ฉด ๋ฉ๋๋ค. ํํฐ๋ง๋ ๊ฒฐ๊ณผ์ ์์ ํญ๋ชฉ์ผ๋ก๋ ํ๋์ค์ด ํ
์คํธ์ ์ฌ์ฉํ ์ ์๋ ๋ค๊ตญ์ด [BERT ๋ชจ๋ธ](https://huggingface.co/nlptown/bert-base-multilingual-uncased-sentiment)์ด ๋ฐํ๋ฉ๋๋ค:
|
|
|
|
```py
|
|
>>> model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
|
|
```
|
|
|
|
[`AutoModelForSequenceClassification`]๊ณผ [`AutoTokenizer`]๋ฅผ ์ฌ์ฉํ์ฌ ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ๊ณผ ๊ด๋ จ๋ ํ ํฌ๋์ด์ ๋ฅผ ๋ก๋ํ์ธ์ (๋ค์ ์น์
์์ [`AutoClass`]์ ๋ํด ๋ ์์ธํ ์์๋ณด๊ฒ ์ต๋๋ค):
|
|
|
|
```py
|
|
>>> from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
|
|
>>> model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
|
>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
```
|
|
|
|
[`pipeline`]์์ ๋ชจ๋ธ๊ณผ ํ ํฌ๋์ด์ ๋ฅผ ์ง์ ํ๋ฉด, ์ด์ `classifier`๋ฅผ ํ๋์ค์ด ํ
์คํธ์ ์ ์ฉํ ์ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
|
>>> classifier("Nous sommes trรจs heureux de vous prรฉsenter la bibliothรจque ๐ค Transformers.")
|
|
[{'label': '5 stars', 'score': 0.7273}]
|
|
```
|
|
|
|
๋ง๋
ํ ๋ชจ๋ธ์ ์ฐพ์ ์ ์๋ ๊ฒฝ์ฐ ๋ฐ์ดํฐ๋ฅผ ๊ธฐ๋ฐ์ผ๋ก ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ์ ๋ฏธ์ธ์กฐ์ ํด์ผ ํฉ๋๋ค. ๋ฏธ์ธ์กฐ์ ๋ฐฉ๋ฒ์ ๋ํ ์์ธํ ๋ด์ฉ์ [๋ฏธ์ธ์กฐ์ ํํ ๋ฆฌ์ผ](./training)์ ์ฐธ์กฐํ์ธ์. ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ์ ๋ฏธ์ธ์กฐ์ ํ ํ์๋ ๋ชจ๋ธ์ Hub์ ์ปค๋ฎค๋ํฐ์ ๊ณต์ ํ์ฌ ๋จธ์ ๋ฌ๋ ๋ฏผ์ฃผํ์ ๊ธฐ์ฌํด์ฃผ์ธ์! ๐ค
|
|
|
|
## AutoClass [[autoclass]]
|
|
|
|
<Youtube id="AhChOFRegn4"/>
|
|
|
|
[`AutoModelForSequenceClassification`]๊ณผ [`AutoTokenizer`] ํด๋์ค๋ ์์์ ๋ค๋ฃฌ [`pipeline`]์ ๊ธฐ๋ฅ์ ๊ตฌํํ๋ ๋ฐ ์ฌ์ฉ๋ฉ๋๋ค. [AutoClass](./model_doc/auto)๋ ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ์ ์ํคํ
์ฒ๋ฅผ ์ด๋ฆ์ด๋ ๊ฒฝ๋ก์์ ์๋์ผ๋ก ๊ฐ์ ธ์ค๋ '๋ฐ๋ก๊ฐ๊ธฐ'์
๋๋ค. ๊ณผ์
์ ์ ํฉํ `AutoClass`๋ฅผ ์ ํํ๊ณ ํด๋น ์ ์ฒ๋ฆฌ ํด๋์ค๋ฅผ ์ ํํ๊ธฐ๋ง ํ๋ฉด ๋ฉ๋๋ค.
|
|
|
|
์ด์ ์น์
์ ์์ ๋ก ๋์๊ฐ์ [`pipeline`]์ ๊ฒฐ๊ณผ๋ฅผ `AutoClass`๋ฅผ ํ์ฉํด ๋ณต์ ํ๋ ๋ฐฉ๋ฒ์ ์ดํด๋ณด๊ฒ ์ต๋๋ค.
|
|
|
|
### AutoTokenizer [[autotokenizer]]
|
|
|
|
ํ ํฌ๋์ด์ ๋ ํ
์คํธ๋ฅผ ๋ชจ๋ธ์ ์
๋ ฅ์ผ๋ก ์ฌ์ฉํ๊ธฐ ์ํด ์ซ์ ๋ฐฐ์ด ํํ๋ก ์ ์ฒ๋ฆฌํ๋ ์ญํ ์ ๋ด๋นํฉ๋๋ค. ํ ํฐํ ๊ณผ์ ์๋ ๋จ์ด๋ฅผ ์ด๋์์ ๋์์ง, ์ด๋ ์์ค๊น์ง ๋๋์ง์ ๊ฐ์ ์ฌ๋ฌ ๊ท์น๋ค์ด ์์ต๋๋ค (ํ ํฐํ์ ๋ํ ์์ธํ ๋ด์ฉ์ [ํ ํฌ๋์ด์ ์์ฝ](./tokenizer_summary)์ ์ฐธ์กฐํ์ธ์). ๊ฐ์ฅ ์ค์ํ ์ ์ ๋ชจ๋ธ์ด ์ฌ์ ํ๋ จ๋ ๋ชจ๋ธ๊ณผ ๋์ผํ ํ ํฐํ ๊ท์น์ ์ฌ์ฉํ๋๋ก ๋์ผํ ๋ชจ๋ธ ์ด๋ฆ์ผ๋ก ํ ํฌ๋์ด์ ๋ฅผ ์ธ์คํด์คํํด์ผ ํ๋ค๋ ๊ฒ์
๋๋ค.
|
|
|
|
[`AutoTokenizer`]๋ก ํ ํฌ๋์ด์ ๋ฅผ ๋ก๋ํ์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import AutoTokenizer
|
|
|
|
>>> model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
|
|
>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
```
|
|
|
|
ํ
์คํธ๋ฅผ ํ ํฌ๋์ด์ ์ ์ ๋ฌํ์ธ์:
|
|
|
|
```py
|
|
>>> encoding = tokenizer("We are very happy to show you the ๐ค Transformers library.")
|
|
>>> print(encoding)
|
|
{'input_ids': [101, 11312, 10320, 12495, 19308, 10114, 11391, 10855, 10103, 100, 58263, 13299, 119, 102],
|
|
'token_type_ids': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
|
'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}
|
|
```
|
|
|
|
ํ ํฌ๋์ด์ ๋ ๋ค์์ ํฌํจํ ๋์
๋๋ฆฌ๋ฅผ ๋ฐํํฉ๋๋ค:
|
|
|
|
* [input_ids](./glossary#input-ids): ํ ํฐ์ ์ซ์ ํํ.
|
|
* [attention_mask](.glossary#attention-mask): ์ด๋ค ํ ํฐ์ ์ฃผ์๋ฅผ ๊ธฐ์ธ์ฌ์ผ ํ๋์ง๋ฅผ ๋ํ๋
๋๋ค.
|
|
|
|
ํ ํฌ๋์ด์ ๋ ์
๋ ฅ์ ๋ฆฌ์คํธ ํํ๋ก๋ ๋ฐ์ ์ ์์ผ๋ฉฐ, ํ
์คํธ๋ฅผ ํจ๋ฉํ๊ณ ์๋ผ๋ด์ด ์ผ์ ํ ๊ธธ์ด์ ๋ฌถ์์ ๋ฐํํ ์๋ ์์ต๋๋ค:
|
|
|
|
|
|
```py
|
|
>>> pt_batch = tokenizer(
|
|
... ["We are very happy to show you the ๐ค Transformers library.", "We hope you don't hate it."],
|
|
... padding=True,
|
|
... truncation=True,
|
|
... max_length=512,
|
|
... return_tensors="pt",
|
|
... )
|
|
```
|
|
|
|
<Tip>
|
|
|
|
[์ ์ฒ๋ฆฌ](./preprocessing) ํํ ๋ฆฌ์ผ์ ์ฐธ์กฐํ์๋ฉด ํ ํฐํ์ ๋ํ ์์ธํ ์ค๋ช
๊ณผ ํจ๊ป ์ด๋ฏธ์ง, ์ค๋์ค์ ๋ฉํฐ๋ชจ๋ฌ ์
๋ ฅ์ ์ ์ฒ๋ฆฌํ๊ธฐ ์ํ [`AutoImageProcessor`]์ [`AutoFeatureExtractor`], [`AutoProcessor`]์ ์ฌ์ฉ๋ฐฉ๋ฒ๋ ์ ์ ์์ต๋๋ค.
|
|
|
|
</Tip>
|
|
|
|
### AutoModel [[automodel]]
|
|
|
|
๐ค Transformers๋ ์ฌ์ ํ๋ จ๋ ์ธ์คํด์ค๋ฅผ ๊ฐ๋จํ๊ณ ํตํฉ๋ ๋ฐฉ๋ฒ์ผ๋ก ๋ก๋ํ ์ ์์ต๋๋ค. ์ฆ, [`AutoTokenizer`]์ฒ๋ผ [`AutoModel`]์ ๋ก๋ํ ์ ์์ต๋๋ค. ์ ์ผํ ์ฐจ์ด์ ์ ๊ณผ์
์ ์๋ง์ [`AutoModel`]์ ์ ํํด์ผ ํ๋ค๋ ์ ์
๋๋ค. ํ
์คํธ (๋๋ ์ํ์ค) ๋ถ๋ฅ์ ๊ฒฝ์ฐ [`AutoModelForSequenceClassification`]์ ๋ก๋ํด์ผ ํฉ๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import AutoModelForSequenceClassification
|
|
|
|
>>> model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
|
|
>>> pt_model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
|
```
|
|
|
|
<Tip>
|
|
|
|
[`AutoModel`] ํด๋์ค์์ ์ง์ํ๋ ๊ณผ์
์ ๋ํด์๋ [๊ณผ์
์์ฝ](./task_summary)์ ์ฐธ์กฐํ์ธ์.
|
|
|
|
</Tip>
|
|
|
|
์ด์ ์ ์ฒ๋ฆฌ๋ ์
๋ ฅ ๋ฌถ์์ ์ง์ ๋ชจ๋ธ์ ์ ๋ฌํด์ผ ํฉ๋๋ค. ์๋์ฒ๋ผ `**`๋ฅผ ์์ ๋ถ์ฌ ๋์
๋๋ฆฌ๋ฅผ ํ์ด์ฃผ๋ฉด ๋ฉ๋๋ค:
|
|
|
|
```py
|
|
>>> pt_outputs = pt_model(**pt_batch)
|
|
```
|
|
|
|
๋ชจ๋ธ์ ์ต์ข
ํ์ฑํ ํจ์ ์ถ๋ ฅ์ `logits` ์์ฑ์ ๋ด๊ฒจ์์ต๋๋ค. `logits`์ softmax ํจ์๋ฅผ ์ ์ฉํ์ฌ ํ๋ฅ ์ ์ป์ ์ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> from torch import nn
|
|
|
|
>>> pt_predictions = nn.functional.softmax(pt_outputs.logits, dim=-1)
|
|
>>> print(pt_predictions)
|
|
tensor([[0.0021, 0.0018, 0.0115, 0.2121, 0.7725],
|
|
[0.2084, 0.1826, 0.1969, 0.1755, 0.2365]], grad_fn=<SoftmaxBackward0>)
|
|
```
|
|
|
|
<Tip>
|
|
|
|
๋ชจ๋ ๐ค Transformers ๋ชจ๋ธ(PyTorch ๋๋ TensorFlow)์ (softmax์ ๊ฐ์) ์ต์ข
ํ์ฑํ ํจ์ *์ด์ ์* ํ
์๋ฅผ ์ถ๋ ฅํฉ๋๋ค. ์๋ํ๋ฉด ์ต์ข
ํ์ฑํ ํจ์์ ์ถ๋ ฅ์ ์ข
์ข
์์ค ํจ์ ์ถ๋ ฅ๊ณผ ๊ฒฐํฉ๋๊ธฐ ๋๋ฌธ์
๋๋ค. ๋ชจ๋ธ ์ถ๋ ฅ์ ํน์ํ ๋ฐ์ดํฐ ํด๋์ค์ด๋ฏ๋ก IDE์์ ์๋ ์์ฑ๋ฉ๋๋ค. ๋ชจ๋ธ ์ถ๋ ฅ์ ํํ์ด๋ ๋์
๋๋ฆฌ์ฒ๋ผ ๋์ํ๋ฉฐ (์ ์, ์ฌ๋ผ์ด์ค ๋๋ ๋ฌธ์์ด๋ก ์ธ๋ฑ์ฑ ๊ฐ๋ฅ), None์ธ ์์ฑ์ ๋ฌด์๋ฉ๋๋ค.
|
|
|
|
</Tip>
|
|
|
|
### ๋ชจ๋ธ ์ ์ฅํ๊ธฐ [[save-a-model]]
|
|
|
|
๋ฏธ์ธ์กฐ์ ๋ ๋ชจ๋ธ์ ํ ํฌ๋์ด์ ์ ํจ๊ป ์ ์ฅํ๋ ค๋ฉด [`PreTrainedModel.save_pretrained`]๋ฅผ ์ฌ์ฉํ์ธ์:
|
|
|
|
```py
|
|
>>> pt_save_directory = "./pt_save_pretrained"
|
|
>>> tokenizer.save_pretrained(pt_save_directory) # doctest: +IGNORE_RESULT
|
|
>>> pt_model.save_pretrained(pt_save_directory)
|
|
```
|
|
|
|
๋ชจ๋ธ์ ๋ค์ ์ฌ์ฉํ๋ ค๋ฉด [`PreTrainedModel.from_pretrained`]๋ก ๋ชจ๋ธ์ ๋ค์ ๋ก๋ํ์ธ์:
|
|
|
|
```py
|
|
>>> pt_model = AutoModelForSequenceClassification.from_pretrained("./pt_save_pretrained")
|
|
```
|
|
|
|
๐ค Transformers์ ๋ฉ์ง ๊ธฐ๋ฅ ์ค ํ๋๋ ๋ชจ๋ธ์ PyTorch ๋๋ TensorFlow ๋ชจ๋ธ๋ก ์ ์ฅํด๋๋ค๊ฐ ๋ค๋ฅธ ํ๋ ์์ํฌ๋ก ๋ค์ ๋ก๋ํ ์ ์๋ ์ ์
๋๋ค. `from_pt` ๋๋ `from_tf` ๋งค๊ฐ๋ณ์๋ฅผ ์ฌ์ฉํ์ฌ ๋ชจ๋ธ์ ํ ํ๋ ์์ํฌ์์ ๋ค๋ฅธ ํ๋ ์์ํฌ๋ก ๋ณํํ ์ ์์ต๋๋ค:
|
|
|
|
|
|
```py
|
|
>>> from transformers import AutoModel
|
|
|
|
>>> tokenizer = AutoTokenizer.from_pretrained(pt_save_directory)
|
|
>>> pt_model = AutoModelForSequenceClassification.from_pretrained(pt_save_directory, from_pt=True)
|
|
```
|
|
|
|
## ์ปค์คํ
๋ชจ๋ธ ๊ตฌ์ถํ๊ธฐ [[custom-model-builds]]
|
|
|
|
๋ชจ๋ธ์ ๊ตฌ์ฑ ํด๋์ค๋ฅผ ์์ ํ์ฌ ๋ชจ๋ธ์ ๊ตฌ์กฐ๋ฅผ ๋ฐ๊ฟ ์ ์์ต๋๋ค. (์๋์ธต์ด๋ ์ดํ
์
ํค๋์ ์์ ๊ฐ์) ๋ชจ๋ธ์ ์์ฑ์ ๊ตฌ์ฑ์์ ์ง์ ๋๊ธฐ ๋๋ฌธ์
๋๋ค. ์ปค์คํ
๊ตฌ์ฑ ํด๋์ค๋ก ๋ชจ๋ธ์ ๋ง๋ค๋ฉด ์ฒ์๋ถํฐ ์์ํด์ผ ํฉ๋๋ค. ๋ชจ๋ธ ์์ฑ์ ๋ฌด์์๋ก ์ด๊ธฐํ๋๋ฏ๋ก ์๋ฏธ ์๋ ๊ฒฐ๊ณผ๋ฅผ ์ป์ผ๋ ค๋ฉด ๋จผ์ ๋ชจ๋ธ์ ํ๋ จ์์ผ์ผ ํฉ๋๋ค.
|
|
|
|
๋จผ์ [`AutoConfig`]๋ฅผ ๊ฐ์ ธ์ค๊ณ ์์ ํ๊ณ ์ถ์ ์ฌ์ ํ์ต๋ ๋ชจ๋ธ์ ๋ก๋ํ์ธ์. [`AutoConfig.from_pretrained`] ๋ด๋ถ์์ (์ดํ
์
ํค๋ ์์ ๊ฐ์ด) ๋ณ๊ฒฝํ๋ ค๋ ์์ฑ๋ฅผ ์ง์ ํ ์ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import AutoConfig
|
|
|
|
>>> my_config = AutoConfig.from_pretrained("distilbert/distilbert-base-uncased", n_heads=12)
|
|
```
|
|
|
|
[`AutoModel.from_config`]๋ฅผ ์ฌ์ฉํ์ฌ ๋ฐ๊พผ ๊ตฌ์ฑ๋๋ก ๋ชจ๋ธ์ ์์ฑํ์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import AutoModel
|
|
|
|
>>> my_model = AutoModel.from_config(my_config)
|
|
```
|
|
|
|
์ปค์คํ
๊ตฌ์ฑ์ ๋ํ ์์ธํ ๋ด์ฉ์ [์ปค์คํ
์ํคํ
์ฒ ๋ง๋ค๊ธฐ](./create_a_model) ๊ฐ์ด๋๋ฅผ ํ์ธํ์ธ์.
|
|
|
|
## Trainer - PyTorch์ ์ต์ ํ๋ ํ๋ จ ๋ฃจํ [[trainer-a-pytorch-optimized-training-loop]]
|
|
|
|
๋ชจ๋ ๋ชจ๋ธ์ [`torch.nn.Module`](https://pytorch.org/docs/stable/nn.html#torch.nn.Module)์ด๋ฏ๋ก ์ผ๋ฐ์ ์ธ ํ๋ จ ๋ฃจํ์์ ์ฌ์ฉํ ์ ์์ต๋๋ค. ์ง์ ํ๋ จ ๋ฃจํ๋ฅผ ์์ฑํ ์๋ ์์ง๋ง, ๐ค Transformers๋ PyTorch๋ฅผ ์ํ [`Trainer`] ํด๋์ค๋ฅผ ์ ๊ณตํฉ๋๋ค. ์ด ํด๋์ค์๋ ๊ธฐ๋ณธ ํ๋ จ ๋ฃจํ๊ฐ ํฌํจ๋์ด ์์ผ๋ฉฐ ๋ถ์ฐ ํ๋ จ, ํผํฉ ์ ๋ฐ๋ ๋ฑ๊ณผ ๊ฐ์ ๊ธฐ๋ฅ์ ์ถ๊ฐ๋ก ์ ๊ณตํฉ๋๋ค.
|
|
|
|
๊ณผ์
์ ๋ฐ๋ผ ๋ค๋ฅด์ง๋ง ์ผ๋ฐ์ ์ผ๋ก [`Trainer`]์ ๋ค์ ๋งค๊ฐ๋ณ์๋ฅผ ์ ๋ฌํฉ๋๋ค:
|
|
|
|
1. [`PreTrainedModel`] ๋๋ [`torch.nn.Module`](https://pytorch.org/docs/stable/nn.html#torch.nn.Module)๋ก ์์ํฉ๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import AutoModelForSequenceClassification
|
|
|
|
>>> model = AutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased")
|
|
```
|
|
|
|
2. [`TrainingArguments`]๋ ํ์ต๋ฅ , ๋ฐฐ์น ํฌ๊ธฐ, ํ๋ จํ ์ํฌํฌ ์์ ๊ฐ์ ๋ชจ๋ธ ํ์ดํผํ๋ผ๋ฏธํฐ๋ฅผ ํฌํจํฉ๋๋ค. ํ๋ จ ์ธ์๋ฅผ ์ง์ ํ์ง ์์ผ๋ฉด ๊ธฐ๋ณธ๊ฐ์ด ์ฌ์ฉ๋ฉ๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import TrainingArguments
|
|
|
|
>>> training_args = TrainingArguments(
|
|
... output_dir="path/to/save/folder/",
|
|
... learning_rate=2e-5,
|
|
... per_device_train_batch_size=8,
|
|
... per_device_eval_batch_size=8,
|
|
... num_train_epochs=2,
|
|
... )
|
|
```
|
|
|
|
3. ํ ํฌ๋์ด์ , ์ด๋ฏธ์ง ํ๋ก์ธ์, ํน์ง ์ถ์ถ๊ธฐ(feature extractor) ๋๋ ํ๋ก์ธ์์ ์ ์ฒ๋ฆฌ ํด๋์ค๋ฅผ ๋ก๋ํ์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import AutoTokenizer
|
|
|
|
>>> tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-uncased")
|
|
```
|
|
|
|
4. ๋ฐ์ดํฐ์
์ ๋ก๋ํ์ธ์:
|
|
|
|
```py
|
|
>>> from datasets import load_dataset
|
|
|
|
>>> dataset = load_dataset("rotten_tomatoes") # doctest: +IGNORE_RESULT
|
|
```
|
|
|
|
5. ๋ฐ์ดํฐ์
์ ํ ํฐํํ๋ ํจ์๋ฅผ ์์ฑํ์ธ์:
|
|
|
|
```py
|
|
>>> def tokenize_dataset(dataset):
|
|
... return tokenizer(dataset["text"])
|
|
```
|
|
|
|
๊ทธ๋ฆฌ๊ณ [`~datasets.Dataset.map`]๋ก ๋ฐ์ดํฐ์
์ ์ฒด์ ์ ์ฉํ์ธ์:
|
|
|
|
```py
|
|
>>> dataset = dataset.map(tokenize_dataset, batched=True)
|
|
```
|
|
|
|
6. [`DataCollatorWithPadding`]์ ์ฌ์ฉํ์ฌ ๋ฐ์ดํฐ์
์ ํ๋ณธ ๋ฌถ์์ ๋ง๋์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import DataCollatorWithPadding
|
|
|
|
>>> data_collator = DataCollatorWithPadding(tokenizer=tokenizer)
|
|
```
|
|
|
|
์ด์ ์์ ๋ชจ๋ ํด๋์ค๋ฅผ [`Trainer`]๋ก ๋ชจ์ผ์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import Trainer
|
|
|
|
>>> trainer = Trainer(
|
|
... model=model,
|
|
... args=training_args,
|
|
... train_dataset=dataset["train"],
|
|
... eval_dataset=dataset["test"],
|
|
... processing_class=tokenizer,
|
|
... data_collator=data_collator,
|
|
... ) # doctest: +SKIP
|
|
```
|
|
|
|
์ค๋น๊ฐ ๋์์ผ๋ฉด [`~Trainer.train`]์ ํธ์ถํ์ฌ ํ๋ จ์ ์์ํ์ธ์:
|
|
|
|
```py
|
|
>>> trainer.train() # doctest: +SKIP
|
|
```
|
|
|
|
<Tip>
|
|
|
|
๋ฒ์ญ์ด๋ ์์ฝ๊ณผ ๊ฐ์ด ์ํ์ค-์ํ์ค ๋ชจ๋ธ์ ์ฌ์ฉํ๋ ๊ณผ์
์๋ [`Seq2SeqTrainer`] ๋ฐ [`Seq2SeqTrainingArguments`] ํด๋์ค๋ฅผ ์ฌ์ฉํ์ธ์.
|
|
|
|
</Tip>
|
|
|
|
[`Trainer`] ๋ด์ ๋ฉ์๋๋ฅผ ์๋ธํด๋์คํํ์ฌ ํ๋ จ ๋ฃจํ๋ฅผ ๋ฐ๊ฟ ์๋ ์์ต๋๋ค. ์ด๋ฌ๋ฉด ์์ค ํจ์, ์ตํฐ๋ง์ด์ , ์ค์ผ์ค๋ฌ์ ๊ฐ์ ๊ธฐ๋ฅ ๋ํ ๋ฐ๊ฟ ์ ์๊ฒ ๋ฉ๋๋ค. ๋ณ๊ฒฝ ๊ฐ๋ฅํ ๋ฉ์๋์ ๋ํด์๋ [`Trainer`] ๋ฌธ์๋ฅผ ์ฐธ๊ณ ํ์ธ์.
|
|
|
|
ํ๋ จ ๋ฃจํ๋ฅผ ์์ ํ๋ ๋ค๋ฅธ ๋ฐฉ๋ฒ์ [Callbacks](./main_classes/callback)๋ฅผ ์ฌ์ฉํ๋ ๊ฒ์
๋๋ค. Callbacks๋ก ๋ค๋ฅธ ๋ผ์ด๋ธ๋ฌ๋ฆฌ์ ํตํฉํ๊ณ , ํ๋ จ ๋ฃจํ๋ฅผ ์ฒดํฌํ์ฌ ์งํ ์ํฉ์ ๋ณด๊ณ ๋ฐ๊ฑฐ๋, ํ๋ จ์ ์กฐ๊ธฐ์ ์ค๋จํ ์ ์์ต๋๋ค. Callbacks์ ํ๋ จ ๋ฃจํ ์์ฒด๋ฅผ ๋ฐ๊พธ์ง๋ ์์ต๋๋ค. ์์ค ํจ์์ ๊ฐ์ ๊ฒ์ ๋ฐ๊พธ๋ ค๋ฉด [`Trainer`]๋ฅผ ์๋ธํด๋์คํํด์ผ ํฉ๋๋ค.
|
|
|
|
## TensorFlow๋ก ํ๋ จ์ํค๊ธฐ [[train-with-tensorflow]]
|
|
|
|
๋ชจ๋ ๋ชจ๋ธ์ [`tf.keras.Model`](https://www.tensorflow.org/api_docs/python/tf/keras/Model)์ด๋ฏ๋ก [Keras](https://keras.io/) API๋ฅผ ํตํด TensorFlow์์ ํ๋ จ์ํฌ ์ ์์ต๋๋ค. ๐ค Transformers๋ ๋ฐ์ดํฐ์
์ ์ฝ๊ฒ `tf.data.Dataset` ํํ๋ก ์ฝ๊ฒ ๋ก๋ํ ์ ์๋ [`~TFPreTrainedModel.prepare_tf_dataset`] ๋ฉ์๋๋ฅผ ์ ๊ณตํ๊ธฐ ๋๋ฌธ์, Keras์ [`compile`](https://keras.io/api/models/model_training_apis/#compile-method) ๋ฐ [`fit`](https://keras.io/api/models/model_training_apis/#fit-method) ๋ฉ์๋๋ก ๋ฐ๋ก ํ๋ จ์ ์์ํ ์ ์์ต๋๋ค.
|
|
|
|
1. [`TFPreTrainedModel`] ๋๋ [`tf.keras.Model`](https://www.tensorflow.org/api_docs/python/tf/keras/Model)๋ก ์์ํฉ๋๋ค:
|
|
|
|
```py
|
|
>>> from transformers import TFAutoModelForSequenceClassification
|
|
|
|
>>> model = TFAutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased")
|
|
```
|
|
|
|
2. ํ ํฌ๋์ด์ , ์ด๋ฏธ์ง ํ๋ก์ธ์, ํน์ง ์ถ์ถ๊ธฐ(feature extractor) ๋๋ ํ๋ก์ธ์์ ๊ฐ์ ์ ์ฒ๋ฆฌ ํด๋์ค๋ฅผ ๋ก๋ํ์ธ์:
|
|
|
|
```py
|
|
>>> from transformers import AutoTokenizer
|
|
|
|
>>> tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-uncased")
|
|
```
|
|
|
|
3. ๋ฐ์ดํฐ์
์ ํ ํฐํํ๋ ํจ์๋ฅผ ์์ฑํ์ธ์:
|
|
|
|
```py
|
|
>>> def tokenize_dataset(dataset):
|
|
... return tokenizer(dataset["text"]) # doctest: +SKIP
|
|
```
|
|
|
|
4. [`~datasets.Dataset.map`]์ ์ฌ์ฉํ์ฌ ์ ์ฒด ๋ฐ์ดํฐ์
์ ํ ํฐํ ํจ์๋ฅผ ์ ์ฉํ๊ณ , ๋ฐ์ดํฐ์
๊ณผ ํ ํฌ๋์ด์ ๋ฅผ [`~TFPreTrainedModel.prepare_tf_dataset`]์ ์ ๋ฌํ์ธ์. ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ๋ณ๊ฒฝํ๊ฑฐ๋ ๋ฐ์ดํฐ์
์ ์์ ์๋ ์์ต๋๋ค:
|
|
|
|
```py
|
|
>>> dataset = dataset.map(tokenize_dataset) # doctest: +SKIP
|
|
>>> tf_dataset = model.prepare_tf_dataset(
|
|
... dataset["train"], batch_size=16, shuffle=True, tokenizer=tokenizer
|
|
... ) # doctest: +SKIP
|
|
```
|
|
|
|
5. ์ค๋น๋์์ผ๋ฉด `compile` ๋ฐ `fit`๋ฅผ ํธ์ถํ์ฌ ํ๋ จ์ ์์ํ์ธ์. ๐ค Transformers์ ๋ชจ๋ ๋ชจ๋ธ์ ๊ณผ์
๊ณผ ๊ด๋ จ๋ ๊ธฐ๋ณธ ์์ค ํจ์๋ฅผ ๊ฐ์ง๊ณ ์์ผ๋ฏ๋ก ๋ช
์์ ์ผ๋ก ์ง์ ํ์ง ์์๋ ๋ฉ๋๋ค:
|
|
|
|
```py
|
|
>>> from tensorflow.keras.optimizers import Adam
|
|
|
|
>>> model.compile(optimizer=Adam(3e-5)) # No loss argument!
|
|
>>> model.fit(tf_dataset) # doctest: +SKIP
|
|
```
|
|
|
|
## ๋ค์ ๋จ๊ณ๋ ๋ฌด์์ธ๊ฐ์? [[whats-next]]
|
|
|
|
๐ค Transformers ๋๋ฌ๋ณด๊ธฐ๋ฅผ ๋ชจ๋ ์ฝ์ผ์
จ๋ค๋ฉด, ๊ฐ์ด๋๋ฅผ ์ดํด๋ณด๊ณ ๋ ๊ตฌ์ฒด์ ์ธ ๊ฒ์ ์ํํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์. ์ด๋ฅผํ
๋ฉด ์ปค์คํ
๋ชจ๋ธ ๊ตฌ์ถํ๋ ๋ฐฉ๋ฒ, ๊ณผ์
์ ์๋ง๊ฒ ๋ชจ๋ธ์ ๋ฏธ์ธ์กฐ์ ํ๋ ๋ฐฉ๋ฒ, ์คํฌ๋ฆฝํธ๋ก ๋ชจ๋ธ ํ๋ จํ๋ ๋ฐฉ๋ฒ ๋ฑ์ด ์์ต๋๋ค. ๐ค Transformers ํต์ฌ ๊ฐ๋
์ ๋ํด ๋ ์์๋ณด๋ ค๋ฉด ์ปคํผ ํ ์ ๋ค๊ณ ๊ฐ๋
๊ฐ์ด๋๋ฅผ ์ดํด๋ณด์ธ์!
|