File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
examples/text_classification/pretrained_models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -113,8 +113,8 @@ def do_train():
113
113
train_ds , dev_ds = load_dataset ("chnsenticorp" , splits = ["train" , "dev" ])
114
114
115
115
# If you wanna use bert/roberta/electra pretrained model,
116
- # model = ppnlp.transformers.BertForSequenceClassification.from_pretrained('bert-base-chinese', num_class =2)
117
- # model = ppnlp.transformers.RobertaForSequenceClassification.from_pretrained('roberta-wwm-ext', num_class =2)
116
+ # model = ppnlp.transformers.BertForSequenceClassification.from_pretrained('bert-base-chinese', num_classes =2)
117
+ # model = ppnlp.transformers.RobertaForSequenceClassification.from_pretrained('roberta-wwm-ext', num_classes =2)
118
118
# model = ppnlp.transformers.ElectraForSequenceClassification.from_pretrained('chinese-electra-small', num_classes=2)
119
119
model = ppnlp .transformers .ErnieForSequenceClassification .from_pretrained (
120
120
'ernie-tiny' , num_classes = len (train_ds .label_list ))
You can’t perform that action at this time.
0 commit comments