修改了两处bug

Author: mathCrazyy

__pycache__/model.cpython-36.pyc

@@ -14,7 +14,7 @@
 class Config(object):
     def __init__(self):
         self.model_name="lm_model"
-        self.data_ori="E:/data/word_nlp/cnews_data/"
+        self.data_ori="/mnt/data3/wuchunsheng/data_all/data_mine/lm_data/"
         self.train_path="train_0.csv"
         self.valid_path="train_0.csv"
         self.test_path="test_100.csv"
@@ -37,6 +37,7 @@ def __init__(self):
         self.train_len=0
         self.test_len = 0
         self.valid_len = 0
+        self.mode="train"
 #data_path="E:/study_series/2020_3/re_write_classify/data/"
 #data_path="/mnt/data3/wuchunsheng/code/nlper/NLP_task/text_classification/my_classification_cnews/2020_3_30/text_classify/data/"
 
@@ -50,4 +51,4 @@ def __init__(self):
 
 train(config,model,train_iter, valid_iter, test_iter)
 
-test(config,model,TEXT,  test_iter)## 测试的是一个正批量的
+test(config,model,TEXT,  test_iter)## 测试的是一个正批量的

__pycache__/train_eval.cpython-36.pyc

@@ -16,7 +16,7 @@
 class Config(object):
     def __init__(self):
         self.model_name = "lm_model"
-        self.data_ori =  "E:/data/word_nlp/cnews_data/"
+        self.data_ori =  "/mnt/data3/wuchunsheng/data_all/data_mine/lm_data/"
         self.train_path = "train_0.csv"
         self.valid_path = "train_0.csv"
         self.test_path = "test_100.csv"

__pycache__/utils.cpython-36.pyc

@@ -72,7 +72,7 @@ def test(config, model, TEXT, test_iter):
 
 
 def test_sentence(config, model, TEXT, sentence):
-    inputs = torch.Tensor([TEXT.vocab[one] for one in sentence]).long().to(config.device)
+    inputs = torch.Tensor([TEXT.vocab.stoi[one] for one in sentence]).long().to(config.device)
     inputs = inputs.view(-1, 1)
     # print(inputs.shape)
     arrs = model(inputs)