dssm train

recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -51,7 +51,7 @@ logger = logging.getLogger(__name__)
 import json
 
 class InferenceFetchHandler(FetchHandler):
-    def __init__(self, var_dict, output_file, batch_size=1000):
+    def __init__(self, var_dict, batch_size=1000):
         super().__init__(var_dict=var_dict, period_secs=1)
         # self.output_file = output_file
         # self.batch_size = batch_size