Update README.md

README.md

@@ -9,9 +9,9 @@ tags:
 - zero-shot
 ---
 
-# Erlangshen-Albert-235M-UniMC-English
+# Erlangshen-UniMC-Albert-235M-English
 - Paper: [Zero-Shot Learners for Nature Language Understanding via a Unified Multiple Choice Perspective](https://github.com/IDEA-CCNL/Fengshenbang-LM)
-- Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
+- Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM/tree/main/fengshen/examples/unimc/)
 - Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
 
 ## 简介 Brief Introduction
@@ -49,43 +49,36 @@ avoiding problems in commonly used large generative models such as FLAN. It not
 
 ```python3
 import argparse
-from fengshen import UniMCPiplines
-
+from fengshen.pipelines.multiplechoice import UniMCPiplines
 
 total_parser = argparse.ArgumentParser("TASK NAME")
 total_parser = UniMCPiplines.piplines_args(total_parser)
 args = total_parser.parse_args()
-
-args.pretrained_model_path = 'IDEA-CCNL/Erlangshen-Albert-235M-UniMC-English'
-args.language=='english'
-
-train_data = []
-dev_data = []
-test_data = [
-        {"texta": "Linguistics is the scientific study of language, and involves an analysis of language form, language meaning, and language in context. The earliest activities in the documentation and description of language have been attributed to the 4th century BCE Indian grammarian Pāṇini, who wrote a formal description of the Sanskrit language in his Aṣṭādhyāyī .", 
-         "textb": "", 
-         "question": "Based on the paragraph", 
-         "choice": [
-            "we can infer that Form and meaning are the only aspects of language linguistics is concerned with.", 
-            "we can not infer that Form and meaning are the only aspects of language linguistics is concerned with.", 
-            "it is difficult for us to infer that Form and meaning are the only aspects of language linguistics is concerned with."
-            ], 
-         "answer": "we can not infer that Form and meaning are the only aspects of language linguistics is concerned with.", 
-         "label": 1, 
-         "id": 0},
-    ]
-
-model = UniMCPiplines(args)
+    
+pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniMC-Albert-235M-English'
+args.language='english'
+args.learning_rate=2e-5
+args.max_length=512
+args.max_epochs=3
+args.batchsize=8
+args.default_root_dir='./'
+model = UniMCPiplines(args, model_path=pretrained_model_path)
+
+train_data = [] 
+dev_data = [] 
+test_data = [{
+	"texta": "it 's just incredibly dull .",
+	"textb": "",
+	"question": "What is sentiment of follow review?",
+	"choice": ["it's great", "it's terrible"],
+	"answer": "",
+	"label": 0,
+	"id": 19
+}]
 
 if args.train:
-    model.fit(train_data, dev_data)
+	model.fit(train_data, dev_data)
 result = model.predict(test_data)
-for line in result[:20]:
-    print(line)
-
-
-
-
 ```
 
 ## 引用 Citation