update readme and make run.py

README.md

@@ -32,6 +32,8 @@ cd BERT-BiLSTM-CRF-NER/
 python3 setup.py install
 ```
 
+if you do not want to install, you just need clone this project and reference the file of <run.py> to train the model or start the service. 
+
 ## UPDATE:
 - 2019.2.25 Fix some bug for ner service
 - 2019.2.19: add text classification service

run.py

@@ -13,9 +13,6 @@
 from __future__ import print_function
 
 
-# sys.path.append('.')
-
-
 def start_server():
     from bert_base.server import BertServer
     from bert_base.server.helper import get_run_args
@@ -27,15 +24,24 @@ def start_server():
     server.join()
 
 
-def start_client():
-    pass
+def train_ner():
+    import os
+    from bert_base.train.train_helper import get_args_parser
+    from bert_base.train.bert_lstm_ner import train
 
-
-# def train_ner():
-#     from bert_lstm_ner import main
-#     args =
-#     main()
+    args = get_args_parser()
+    if True:
+        import sys
+        param_str = '\n'.join(['%20s = %s' % (k, v) for k, v in sorted(vars(args).items())])
+        print('usage: %s\n%20s   %s\n%s\n%s\n' % (' '.join(sys.argv), 'ARG', 'VALUE', '_' * 50, param_str))
+    print(args)
+    os.environ['CUDA_VISIBLE_DEVICES'] = args.device_map
+    train(args=args)
 
 
 if __name__ == '__main__':
-    start_server()
+    """
+    如果想训练，那么直接 指定参数跑，如果想启动服务，那么注释掉train,打开server即可
+    """
+    train_ner()
+    #start_server()