edumunozsala commited on
Commit
a8c160a
1 Parent(s): 94189dd

Updated README.md

Browse files

Improve text format and clean some useless data

Files changed (1) hide show
  1. README.md +33 -24
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
- dataset:
 
18
  name: "IMDb Reviews in Spanish"
 
19
  type: IMDbreviews_es
20
- metrics:
21
- - name: Accuracy
22
- type: accuracy
 
23
  value: 0.881866
24
- - name: F1 Score
25
- type: f1
 
26
  value: 0.008272
27
- - name: Precision
28
- type: precision
 
29
  value: 0.858605
30
- - name: Recall
31
- type: recall
 
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
- ## Results
62
 
63
  epoch = 1.0
 
64
  eval_accuracy = 0.8629333333333333
 
65
  eval_f1 = 0.8648790746582545
 
66
  eval_loss = 0.3160930573940277
67
- eval_mem_cpu_alloc_delta = 0
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