Update README.md
Browse files
README.md
CHANGED
|
@@ -44,22 +44,15 @@ KorSciDeBERTa는 Microsoft DeBERTa 모델의 아키텍쳐를 기반으로, 논
|
|
| 44 |
|
| 45 |
<pre><code>
|
| 46 |
from tokenization_korscideberta import DebertaV2Tokenizer
|
| 47 |
-
|
| 48 |
from transformers import AutoModelForSequenceClassification
|
| 49 |
|
| 50 |
|
| 51 |
tokenizer = DebertaV2Tokenizer.from_pretrained("kisti/korscideberta")
|
| 52 |
-
|
| 53 |
model = AutoModelForSequenceClassification.from_pretrained("kisti/korscideberta", num_labels=6, hidden_dropout_prob=0.1, attention_probs_dropout_prob=0.1)
|
| 54 |
-
|
| 55 |
#model = AutoModelForMaskedLM.from_pretrained("kisti/korscideberta")
|
| 56 |
-
|
| 57 |
''''''
|
| 58 |
-
|
| 59 |
train_metrics = trainer.train().metrics
|
| 60 |
-
|
| 61 |
trainer.save_metrics("train", train_metrics)
|
| 62 |
-
|
| 63 |
trainer.push_to_hub()
|
| 64 |
|
| 65 |
</code></pre>
|
|
|
|
| 44 |
|
| 45 |
<pre><code>
|
| 46 |
from tokenization_korscideberta import DebertaV2Tokenizer
|
|
|
|
| 47 |
from transformers import AutoModelForSequenceClassification
|
| 48 |
|
| 49 |
|
| 50 |
tokenizer = DebertaV2Tokenizer.from_pretrained("kisti/korscideberta")
|
|
|
|
| 51 |
model = AutoModelForSequenceClassification.from_pretrained("kisti/korscideberta", num_labels=6, hidden_dropout_prob=0.1, attention_probs_dropout_prob=0.1)
|
|
|
|
| 52 |
#model = AutoModelForMaskedLM.from_pretrained("kisti/korscideberta")
|
|
|
|
| 53 |
''''''
|
|
|
|
| 54 |
train_metrics = trainer.train().metrics
|
|
|
|
| 55 |
trainer.save_metrics("train", train_metrics)
|
|
|
|
| 56 |
trainer.push_to_hub()
|
| 57 |
|
| 58 |
</code></pre>
|