Librarian Bot: Add base_model information to model

#2
Files changed (1) hide show
  1. README.md +4 -3
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
- - name: Accuracy
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