Hugging Face's logo

keras
/
roberta_base_en

Text Classification
KerasHub
Keras
English
Model card Files Community
Divyasreepat commited on
Commit
5a0334f
·
verified ·
1 Parent(s): 1288f84

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +5 -8
README.md CHANGED
@@ -1,6 +1,3 @@
1
- ---
2
- library_name: keras-hub
3
- ---
4
  ### Model Overview
5
  A RoBERTa encoder network.
6
 
@@ -50,7 +47,7 @@ labels = [0, 3]
50
 
51
  # Pretrained classifier.
52
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
53
- "roberta_base_en",
54
  num_classes=4,
55
  )
56
  classifier.fit(x=features, y=labels, batch_size=2)
@@ -78,14 +75,14 @@ labels = [0, 3]
78
 
79
  # Pretrained classifier without preprocessing.
80
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
81
- "roberta_base_en",
82
  num_classes=4,
83
  preprocessor=None,
84
  )
85
  classifier.fit(x=features, y=labels, batch_size=2)
86
  ```
87
 
88
- ## Example Usage with Hugging Face URI
89
 
90
  ```python
91
  import keras
@@ -100,7 +97,7 @@ labels = [0, 3]
100
 
101
  # Pretrained classifier.
102
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
103
- "hf://keras/roberta_base_en",
104
  num_classes=4,
105
  )
106
  classifier.fit(x=features, y=labels, batch_size=2)
@@ -128,7 +125,7 @@ labels = [0, 3]
128
 
129
  # Pretrained classifier without preprocessing.
130
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
131
- "hf://keras/roberta_base_en",
132
  num_classes=4,
133
  preprocessor=None,
134
  )
 
 
 
 
1
  ### Model Overview
2
  A RoBERTa encoder network.
3
 
 
47
 
48
  # Pretrained classifier.
49
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
50
+ "${VARIATION_SLUG}",
51
  num_classes=4,
52
  )
53
  classifier.fit(x=features, y=labels, batch_size=2)
 
75
 
76
  # Pretrained classifier without preprocessing.
77
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
78
+ "${VARIATION_SLUG}",
79
  num_classes=4,
80
  preprocessor=None,
81
  )
82
  classifier.fit(x=features, y=labels, batch_size=2)
83
  ```
84
 
85
+ ## Example Usage with HuggingFace uri
86
 
87
  ```python
88
  import keras
 
97
 
98
  # Pretrained classifier.
99
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
100
+ "${VARIATION_SLUG}",
101
  num_classes=4,
102
  )
103
  classifier.fit(x=features, y=labels, batch_size=2)
 
125
 
126
  # Pretrained classifier without preprocessing.
127
  classifier = keras_nlp.models.RobertaClassifier.from_preset(
128
+ "${VARIATION_SLUG}",
129
  num_classes=4,
130
  preprocessor=None,
131
  )