罗俊辉 1 ano atrás
pai
commit
ba8da7b4d8
1 arquivos alterados com 2 adições e 1 exclusões
  1. 2 1
      applications/functions.py

+ 2 - 1
applications/functions.py

@@ -39,7 +39,7 @@ class ParamProcess(object):
                 features['tag3'] = None
         df = pd.DataFrame([features])
         print("data_frame", df.columns)
-        df.drop('title', axis=1)
+        df = df.drop('title', axis=1)
         return df
 
     async def predict_score(self, version, features):
@@ -55,6 +55,7 @@ class ParamProcess(object):
                 result = list(result)
                 return result
             case "v2":
+                print("predict_data", features)
                 result = await self.model_v2.predict(features)
                 print("predict", list(result))
                 result = list(result)