Ver código fonte

generate label for mysql

罗俊辉 1 ano atrás
pai
commit
e8bdf26e9a
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      main_userupload.py

+ 1 - 1
main_userupload.py

@@ -74,7 +74,7 @@ class LightGBM(object):
         df = df.dropna(subset=['label'])
         labels = df['label']
         temp = sorted(labels)
-        yc = temp[int(len(temp) * 0.7)]
+        yc = temp[int(len(temp) * 0.8)]
         print("阈值", yc)
         labels = [0 if i < yc else 1 for i in labels]
         features = df.drop("label", axis=1)