修改了一些提示和内存释放

run_cnn.py

@@ -17,6 +17,8 @@ from data.cnews_loader import read_vocab, read_category, batch_iter, process_fil
 import pandas as pd
 from function import Index
 
+os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
+
 base_dir = 'data/name2category'
 train_dir = os.path.join(base_dir, 'name2category.train.txt')
 test_dir = os.path.join(base_dir, 'name2category.test.txt')
@@ -202,7 +204,7 @@ class name2subcategory():
     def namelyst_predict(self, contents):
         from tensorflow import keras as kr
 
-        print("Loading predicted data...")
+        #print("Loading predicted data...")
         data_id = []
         for i in range(len(contents)):
             data_id.append([self.word_to_id[x] for x in contents[i] if x in self.word_to_id])

示例.py

 from run_cnn import name2subcategory
 
-name_list = ['安赛瑞 消防斧 应急斧头破拆斧工具 消防器材斧子消防站配置 消防检查斧子 25847'] 
 a = name2subcategory()
-category = a.namelyst_predict(name_list)
\ No newline at end of file
+while True:
+    name = input('请输入产品名称:')
+    name_list = [name]
+    category = a.namelyst_predict(name_list)
+    print(category[0])