service embedding

train/train_service.py

@@ -8,7 +8,6 @@ from datetime import date, timedelta
 
 
 sys.path.append(os.path.dirname(os.path.abspath(os.path.dirname(__file__))))
-import utils.connUtils as connUtils
 import utils.configUtils as configUtils
 
 ITEM_NUMBER_COLUMNS = ["item_"+c for c in ["smart_rank2"]]
@@ -34,21 +33,6 @@ def is_float(s):
     except ValueError:
         return False
 
-#数据字典
-def getDataVocabFromRedis(version):
-    conn = connUtils.getRedisConn()
-    key = "Strategy:rec:vocab:service:"+version
-    dataVocabStr = conn.get(key)
-    if dataVocabStr:
-        dataVocab = json.loads(str(dataVocabStr, encoding="utf-8"),encoding='utf-8')
-        print("-----data_vocab-----")
-        for k, v in dataVocab.items():
-            print(k, len(v))
-    else:
-        dataVocab = None
-
-    return dataVocab
-
 # 数据类型转换
 def csvTypeConvert(columns,df,data_vocab):
     df["label"] = df["label"].astype("int")

utils/connUtils.py

 import redis
 
-def getRedisConn():
+def getRedisConn4():
     pool = redis.ConnectionPool(host="172.16.50.145",password="XfkMCCdWDIU%ls$h",port=6379,db=0)
     conn = redis.Redis(connection_pool=pool)
     # conn = redis.Redis(host="172.16.50.145", port=6379, password="XfkMCCdWDIU%ls$h",db=0)
     # conn = redis.Redis(host="172.18.51.10", port=6379,db=0) #test
     return conn
 
+def getRedisConn():
+    pool = redis.ConnectionPool(host="172.16.40.133",password="ReDis!GmTx*0aN6",port=6379,db=0)
+    conn = redis.Redis(connection_pool=pool)
+    # conn = redis.Redis(host="172.16.50.145", port=6379, password="XfkMCCdWDIU%ls$h",db=0)
+    # conn = redis.Redis(host="172.18.51.10", port=6379,db=0) #test
+    return conn
+
 
 if __name__ == '__main__':
     # REDIS_URL = 'redis://:@172.18.51.10:6379'