Instructions to use Kate-lf/emotion-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Kate-lf/emotion-classification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Kate-lf/emotion-classification")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Kate-lf/emotion-classification") model = AutoModelForSequenceClassification.from_pretrained("Kate-lf/emotion-classification", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress, epoch 1
Browse files- config.json +16 -12
- model.safetensors +2 -2
- tokenizer.json +6 -1
- training_args.bin +1 -1
config.json
CHANGED
|
@@ -10,22 +10,26 @@
|
|
| 10 |
"hidden_dropout_prob": 0.1,
|
| 11 |
"hidden_size": 768,
|
| 12 |
"id2label": {
|
| 13 |
-
"0": "
|
| 14 |
-
"1": "
|
| 15 |
-
"2": "
|
| 16 |
-
"3": "
|
| 17 |
-
"4": "
|
| 18 |
-
"5": "
|
|
|
|
|
|
|
| 19 |
},
|
| 20 |
"initializer_range": 0.02,
|
| 21 |
"intermediate_size": 3072,
|
| 22 |
"label2id": {
|
| 23 |
-
"
|
| 24 |
-
"
|
| 25 |
-
"
|
| 26 |
-
"
|
| 27 |
-
"
|
| 28 |
-
"
|
|
|
|
|
|
|
| 29 |
},
|
| 30 |
"layer_norm_eps": 1e-12,
|
| 31 |
"max_position_embeddings": 512,
|
|
|
|
| 10 |
"hidden_dropout_prob": 0.1,
|
| 11 |
"hidden_size": 768,
|
| 12 |
"id2label": {
|
| 13 |
+
"0": "\u4f24\u5fc3",
|
| 14 |
+
"1": "\u5173\u5fc3",
|
| 15 |
+
"2": "\u538c\u6076",
|
| 16 |
+
"3": "\u5e73\u9759",
|
| 17 |
+
"4": "\u60ca\u8bb6",
|
| 18 |
+
"5": "\u5f00\u5fc3",
|
| 19 |
+
"6": "\u751f\u6c14",
|
| 20 |
+
"7": "\u7591\u95ee"
|
| 21 |
},
|
| 22 |
"initializer_range": 0.02,
|
| 23 |
"intermediate_size": 3072,
|
| 24 |
"label2id": {
|
| 25 |
+
"\u4f24\u5fc3": 0,
|
| 26 |
+
"\u5173\u5fc3": 1,
|
| 27 |
+
"\u538c\u6076": 2,
|
| 28 |
+
"\u5e73\u9759": 3,
|
| 29 |
+
"\u5f00\u5fc3": 5,
|
| 30 |
+
"\u60ca\u8bb6": 4,
|
| 31 |
+
"\u751f\u6c14": 6,
|
| 32 |
+
"\u7591\u95ee": 7
|
| 33 |
},
|
| 34 |
"layer_norm_eps": 1e-12,
|
| 35 |
"max_position_embeddings": 512,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a70c1584f6cb7fa4566d5e18e5248ca74b326db355762cfaa55d0abb4c9be679
|
| 3 |
+
size 409118696
|
tokenizer.json
CHANGED
|
@@ -1,6 +1,11 @@
|
|
| 1 |
{
|
| 2 |
"version": "1.0",
|
| 3 |
-
"truncation":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
"padding": null,
|
| 5 |
"added_tokens": [
|
| 6 |
{
|
|
|
|
| 1 |
{
|
| 2 |
"version": "1.0",
|
| 3 |
+
"truncation": {
|
| 4 |
+
"direction": "Right",
|
| 5 |
+
"max_length": 512,
|
| 6 |
+
"strategy": "LongestFirst",
|
| 7 |
+
"stride": 0
|
| 8 |
+
},
|
| 9 |
"padding": null,
|
| 10 |
"added_tokens": [
|
| 11 |
{
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5841
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7cd447230fa5f24fd4d317f21bcfe73c78e26b813d1cd99d1fc1576829aa2524
|
| 3 |
size 5841
|