精排序

src/service/recommend_service.py

@@ -16,7 +16,6 @@ from ydl_ai_recommender.src.core.recommender import UserCFRecommender
 
 logger = create_logger(__name__, 'service.log', is_rotating=True)
 recommender = UserCFRecommender(top_n=2, k=50, u2c='order')
-recommenderByXgb = RecommendByXgboost()
 
 class RecommendHandler(tornado.web.RequestHandler):
     executor = ThreadPoolExecutor(1)
@@ -74,63 +73,6 @@ class RecommendHandler(tornado.web.RequestHandler):
         logger.info('response@@uid=%s@@ret=%s', uid, ret_str)
         return ret
 
-class RecommendXgbHandler(tornado.web.RequestHandler):
-    executor = ThreadPoolExecutor(1)
-
-    @tornado.gen.coroutine
-    def get(self):
-        uid = self.get_argument('uid', None)
-        if uid is None:
-            logger.warn('请求参数不正确，无uid')
-
-        size = self.get_argument('size', 100)
-        try:
-            size = int(size)
-        except Exception as e:
-            logger.warn('size=%s 不是数字', size)
-            size = 100
-
-        ret = yield self.run(uid, size)
-        self.write(ret)
-
-
-    @tornado.gen.coroutine
-    def post(self):
-        param = json.loads(self.request.body.decode('utf-8'))
-        uid = param.get('uid', None)
-        size = param.get('size', 100)
-        if uid is None:
-            logger.warn('请求参数不正确，无uid')
-
-        ret = yield self.run(uid, size)
-        self.write(ret)
-
-
-    @run_on_executor
-    def run(self, uid, size=100):
-        logger.info('request@@uid=%s@@size=%s', uid, size)
-        try:
-            start_time = time.time()
-            recommend_result = recommenderByXgb.recommend(uid, size=size, is_merge=True)
-            logger.info('request@@uid=%s@@size=%s, cost %s ms', uid, size, (time.time()-start_time)*1000)
-            ret = {
-                'status': 'success',
-                'code': 0,
-                'data': recommend_result,
-                'total_count': len(recommend_result),
-            }
-        except Exception as e:
-            logger.error('执行推荐函数报错', exc_info=True)
-            ret = {
-                'status': 'error',
-                'code': 1,
-                'data': [],
-                'total_count': 0,
-            }
-        ret_str = json.dumps(ret, ensure_ascii=False)
-        logger.info('response@@uid=%s@@ret=%s', uid, ret_str)
-        return ret
-
 if __name__ == '__main__':
 
     tornado.options.define('port', default=8868, type=int, help='服务启动的端口号')