rafaelm47labs
commited on
Commit
•
9f34677
1
Parent(s):
e372cb5
Update README.md
Browse files
README.md
CHANGED
@@ -1,8 +1,29 @@
|
|
1 |
-
** This is work in progress **
|
2 |
|
3 |
-
Spanish News Classification Headlines
|
4 |
|
5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
'sociedad'), the arquitecture of the NN is only in Pytorch.
|
7 |
-
|
|
|
|
|
8 |
An example on how to use the model in this page can be found in this colab notebook: https://colab.research.google.com/drive/1XsKea6oMyEckye2FePW_XN7Rf8v41Cw_?usp=sharing
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
|
|
|
2 |
|
3 |
+
# Spanish News Classification Headlines
|
4 |
+
|
5 |
+
SNCH: this model was develop by [M47Labs](https://www.m47labs.com/es/) the goal is sequence classification,the base model use was [BETO](https://huggingface.co/dccuchile/bert-base-spanish-wwm-cased)
|
6 |
+
|
7 |
+
|
8 |
+
|
9 |
+
trained on private dataset. The model will output the probability of a label('ciencia_tecnologia','clickbait','cultura','deportes','economia','educacion','medio_ambiente', 'opinion','politica',
|
10 |
'sociedad'), the arquitecture of the NN is only in Pytorch.
|
11 |
+
|
12 |
+
# Example of Use
|
13 |
+
|
14 |
An example on how to use the model in this page can be found in this colab notebook: https://colab.research.google.com/drive/1XsKea6oMyEckye2FePW_XN7Rf8v41Cw_?usp=sharing
|
15 |
+
|
16 |
+
|
17 |
+
# Finetune Hyperparameters
|
18 |
+
|
19 |
+
|
20 |
+
* MAX_LEN = 32
|
21 |
+
|
22 |
+
* TRAIN_BATCH_SIZE = 8
|
23 |
+
|
24 |
+
* VALID_BATCH_SIZE = 4
|
25 |
+
|
26 |
+
* EPOCHS = 5
|
27 |
+
|
28 |
+
* LEARNING_RATE = 1e-05
|
29 |
+
|