Browse Source

feat:添加评估结果分析脚本

zhaohaipeng 6 months ago
parent
commit
10940d3959
1 changed files with 1 additions and 1 deletions
  1. 1 1
      ad/model_predict_analyse.py

+ 1 - 1
ad/model_predict_analyse.py

@@ -17,7 +17,7 @@ def read_predict(hdfs_path: str) -> list:
                         continue
                         continue
                     cid = split[3].split("_")[0]
                     cid = split[3].split("_")[0]
                     label = split[0]
                     label = split[0]
-                    score = split[2].replace("[", "").replace("]", "").split(",")[1]
+                    score = float(split[2].replace("[", "").replace("]", "").split(",")[1])
 
 
                     result.append({
                     result.append({
                         "cid": cid,
                         "cid": cid,