|
@@ -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)
|