全pcs列,即单位

predict/base_handler.py

@@ -175,5 +175,18 @@ class BasePredictor:
             if pat.findall(str(i)):
             if pat.findall(str(i)):
                 count += 1
                 count += 1
         return round(count / len(data), 3) >= REF_LEVEL or False
         return round(count / len(data), 3) >= REF_LEVEL or False
+
+    @classmethod
+    def is_pcs(self, data):
+        """
+        pcs列
+        """
+        PCS_LEVEL = 0.5
+        count = 0
+        for i in data:
+            if str(i).lower().strip() == 'pcs':
+                count += 1
+        return round(count / len(data), 3) >= PCS_LEVEL or False
+
 if __name__ == "__main__":
 if __name__ == "__main__":
     print(BasePredictor.is_num([3400.0, 5920.0, 4849.0, 2544.0, 3270.0, 52751.0, 2031.0, 5302.0, 726.0, 1247.0, 2472.0, 689.0, 6049.0, 26796.0, 6164.0, 1605.0, 4346.0, 640.0, 960.0, 960.0, 320.0, 160.0, 860.0, 160.0, 320.0, 3183.0, 10151.0, 640.0, 130.0, 1237.0, 800.0, 960.0, 3740.0, 17701.0, 2146.0, 1280.0, 160.0, 1120.0, 160.0, 480.0, 960.0, 480.0, 160.0, 4717.0, 160.0, 160.0, 160.0, 640.0, 160.0, 320.0, 160.0, 160.0, 800.0, 800.0, 480.0, 1600.0, 155.0, 960.0, 320.0, 944.0, 160.0, 160.0, 1280.0, 1852.0, 7680.0, 7680.0, 2880.0, 160.0, 224.0, 480.0, 480.0, 640.0, 160.0, 640.0, 320.0, 1760.0, 640.0, 480.0, 960.0, 160.0, 160.0, 160.0, 160.0, 1920.0, 160.0, 5600.0, 480.0, 2560.0, 160.0, 160.0, 160.0, 160.0, 160.0, 1280.0, 160.0, 160.0, 160.0, 160.0, 160.0, 320.0, 0.0, 160.0, 160.0]))
     print(BasePredictor.is_num([3400.0, 5920.0, 4849.0, 2544.0, 3270.0, 52751.0, 2031.0, 5302.0, 726.0, 1247.0, 2472.0, 689.0, 6049.0, 26796.0, 6164.0, 1605.0, 4346.0, 640.0, 960.0, 960.0, 320.0, 160.0, 860.0, 160.0, 320.0, 3183.0, 10151.0, 640.0, 130.0, 1237.0, 800.0, 960.0, 3740.0, 17701.0, 2146.0, 1280.0, 160.0, 1120.0, 160.0, 480.0, 960.0, 480.0, 160.0, 4717.0, 160.0, 160.0, 160.0, 640.0, 160.0, 320.0, 160.0, 160.0, 800.0, 800.0, 480.0, 1600.0, 155.0, 960.0, 320.0, 944.0, 160.0, 160.0, 1280.0, 1852.0, 7680.0, 7680.0, 2880.0, 160.0, 224.0, 480.0, 480.0, 640.0, 160.0, 640.0, 320.0, 1760.0, 640.0, 480.0, 960.0, 160.0, 160.0, 160.0, 160.0, 1920.0, 160.0, 5600.0, 480.0, 2560.0, 160.0, 160.0, 160.0, 160.0, 160.0, 1280.0, 160.0, 160.0, 160.0, 160.0, 160.0, 320.0, 0.0, 160.0, 160.0]))

predict/dict_predict.py

@@ -100,6 +100,9 @@ class DicPredict(BasePredictor):
             if self.is_ref(no_null_v):
             if self.is_ref(no_null_v):
                 ab_result[k] = '位号'
                 ab_result[k] = '位号'
                 continue
                 continue
+            if self.is_pcs(no_null_v):
+                ab_result[k] = '单位'
+                continue
 
 
             if bol:
             if bol:
                 prob_columns.append(k)
                 prob_columns.append(k)