edumunozsala
commited on
Commit
•
a8c160a
1
Parent(s):
94189dd
Updated README.md
Browse filesImprove text format and clean some useless data
README.md
CHANGED
@@ -1,41 +1,55 @@
|
|
1 |
---
|
2 |
language: es
|
|
|
3 |
tags:
|
4 |
- sagemaker
|
5 |
- roberta
|
6 |
- ruperta
|
7 |
- TextClassification
|
|
|
8 |
license: apache-2.0
|
|
|
9 |
datasets:
|
10 |
- IMDbreviews_es
|
|
|
11 |
model-index:
|
12 |
- name: RuPERTa_base_sentiment_analysis_es
|
|
|
13 |
results:
|
14 |
-
- task:
|
|
|
15 |
name: Sentiment Analysis
|
|
|
16 |
type: sentiment-analysis
|
17 |
-
|
|
|
18 |
name: "IMDb Reviews in Spanish"
|
|
|
19 |
type: IMDbreviews_es
|
20 |
-
|
21 |
-
|
22 |
-
|
|
|
23 |
value: 0.881866
|
24 |
-
|
25 |
-
|
|
|
26 |
value: 0.008272
|
27 |
-
|
28 |
-
|
|
|
29 |
value: 0.858605
|
30 |
-
|
31 |
-
|
|
|
32 |
value: 0.920062
|
|
|
33 |
## `RuPERTa_base_sentiment_analysis_es`
|
34 |
|
35 |
This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container.
|
36 |
|
37 |
The base model is RuPERTa-base (uncased) which is a RoBERTa model trained on a uncased version of big Spanish corpus.
|
38 |
-
It was trained by mrm8488, Manuel Romero.
|
39 |
|
40 |
## Hyperparameters
|
41 |
|
@@ -53,23 +67,18 @@ It was trained by mrm8488, Manuel Romero.
|
|
53 |
"train_batch_size": "32",
|
54 |
"train_filename": "\"train_data.pt\"",
|
55 |
"val_filename": "\"val_data.pt\""
|
56 |
-
}
|
57 |
-
|
58 |
-
|
59 |
-
## Usage
|
60 |
|
61 |
-
##
|
62 |
|
63 |
epoch = 1.0
|
|
|
64 |
eval_accuracy = 0.8629333333333333
|
|
|
65 |
eval_f1 = 0.8648790746582545
|
|
|
66 |
eval_loss = 0.3160930573940277
|
67 |
-
|
68 |
-
eval_mem_cpu_peaked_delta = 0
|
69 |
-
eval_mem_gpu_alloc_delta = 0
|
70 |
-
eval_mem_gpu_peaked_delta = 94507520
|
71 |
eval_precision = 0.8479381443298969
|
72 |
-
eval_recall = 0.8825107296137339
|
73 |
-
eval_runtime = 114.4994
|
74 |
-
eval_samples_per_second = 32.751
|
75 |
|
|
|
|
1 |
---
|
2 |
language: es
|
3 |
+
|
4 |
tags:
|
5 |
- sagemaker
|
6 |
- roberta
|
7 |
- ruperta
|
8 |
- TextClassification
|
9 |
+
|
10 |
license: apache-2.0
|
11 |
+
|
12 |
datasets:
|
13 |
- IMDbreviews_es
|
14 |
+
|
15 |
model-index:
|
16 |
- name: RuPERTa_base_sentiment_analysis_es
|
17 |
+
|
18 |
results:
|
19 |
+
- task:
|
20 |
+
|
21 |
name: Sentiment Analysis
|
22 |
+
|
23 |
type: sentiment-analysis
|
24 |
+
- dataset:
|
25 |
+
|
26 |
name: "IMDb Reviews in Spanish"
|
27 |
+
|
28 |
type: IMDbreviews_es
|
29 |
+
|
30 |
+
- metrics:
|
31 |
+
- name: Accuracy,
|
32 |
+
type: accuracy,
|
33 |
value: 0.881866
|
34 |
+
|
35 |
+
- name: F1 Score,
|
36 |
+
type: f1,
|
37 |
value: 0.008272
|
38 |
+
|
39 |
+
- name: Precision,
|
40 |
+
type: precision,
|
41 |
value: 0.858605
|
42 |
+
|
43 |
+
- name: Recall,
|
44 |
+
type: recall,
|
45 |
value: 0.920062
|
46 |
+
|
47 |
## `RuPERTa_base_sentiment_analysis_es`
|
48 |
|
49 |
This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container.
|
50 |
|
51 |
The base model is RuPERTa-base (uncased) which is a RoBERTa model trained on a uncased version of big Spanish corpus.
|
52 |
+
It was trained by mrm8488, Manuel Romero. It is fine-tuned for a sentiment analysis task.
|
53 |
|
54 |
## Hyperparameters
|
55 |
|
|
|
67 |
"train_batch_size": "32",
|
68 |
"train_filename": "\"train_data.pt\"",
|
69 |
"val_filename": "\"val_data.pt\""
|
70 |
+
}
|
|
|
|
|
|
|
71 |
|
72 |
+
## Evaluation results
|
73 |
|
74 |
epoch = 1.0
|
75 |
+
|
76 |
eval_accuracy = 0.8629333333333333
|
77 |
+
|
78 |
eval_f1 = 0.8648790746582545
|
79 |
+
|
80 |
eval_loss = 0.3160930573940277
|
81 |
+
|
|
|
|
|
|
|
82 |
eval_precision = 0.8479381443298969
|
|
|
|
|
|
|
83 |
|
84 |
+
eval_recall = 0.8825107296137339
|