Divyasreepat commited on
Commit
07318b4
·
verified ·
1 Parent(s): f5f8a85

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,7 +50,7 @@ labels = [0, 3]
50
 
51
  # Pretrained classifier.
52
  classifier = keras_hub.models.RobertaClassifier.from_preset(
53
- "${VARIATION_SLUG}",
54
  num_classes=4,
55
  )
56
  classifier.fit(x=features, y=labels, batch_size=2)
@@ -78,7 +78,7 @@ labels = [0, 3]
78
 
79
  # Pretrained classifier without preprocessing.
80
  classifier = keras_hub.models.RobertaClassifier.from_preset(
81
- "${VARIATION_SLUG}",
82
  num_classes=4,
83
  preprocessor=None,
84
  )
 
50
 
51
  # Pretrained classifier.
52
  classifier = keras_hub.models.RobertaClassifier.from_preset(
53
+ "roberta_base_en",
54
  num_classes=4,
55
  )
56
  classifier.fit(x=features, y=labels, batch_size=2)
 
78
 
79
  # Pretrained classifier without preprocessing.
80
  classifier = keras_hub.models.RobertaClassifier.from_preset(
81
+ "roberta_base_en",
82
  num_classes=4,
83
  preprocessor=None,
84
  )