Librarian Bot: Add base_model information to model
#2
by
librarian-bot
- opened
README.md
CHANGED
@@ -6,12 +6,13 @@ datasets:
|
|
6 |
- amazon_us_reviews
|
7 |
metrics:
|
8 |
- accuracy
|
|
|
9 |
model-index:
|
10 |
- name: bert_category_prediction_amazon_book_reviews
|
11 |
results:
|
12 |
- task:
|
13 |
-
name: Text Classification
|
14 |
type: text-classification
|
|
|
15 |
dataset:
|
16 |
name: amazon_us_reviews
|
17 |
type: amazon_us_reviews
|
@@ -19,9 +20,9 @@ model-index:
|
|
19 |
split: train[:100]
|
20 |
args: Books_v1_00
|
21 |
metrics:
|
22 |
-
-
|
23 |
-
type: accuracy
|
24 |
value: 0.6
|
|
|
25 |
---
|
26 |
|
27 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
6 |
- amazon_us_reviews
|
7 |
metrics:
|
8 |
- accuracy
|
9 |
+
base_model: distilbert-base-uncased
|
10 |
model-index:
|
11 |
- name: bert_category_prediction_amazon_book_reviews
|
12 |
results:
|
13 |
- task:
|
|
|
14 |
type: text-classification
|
15 |
+
name: Text Classification
|
16 |
dataset:
|
17 |
name: amazon_us_reviews
|
18 |
type: amazon_us_reviews
|
|
|
20 |
split: train[:100]
|
21 |
args: Books_v1_00
|
22 |
metrics:
|
23 |
+
- type: accuracy
|
|
|
24 |
value: 0.6
|
25 |
+
name: Accuracy
|
26 |
---
|
27 |
|
28 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|