预测代码

main.py

@@ -84,6 +84,7 @@ class LightGBM(object):
         index_t = int(len(y_list) * self.split_c)
         temp = sorted(y_list)
         yuzhi = temp[int(len(temp) * self.yc) - 1]
+        print("阈值是: {}".format(yuzhi))
         y__list = [0 if i <= yuzhi else 1 for i in y_list]
         y_train = np.array(y__list[:index_t])
         y_test = np.array(y__list[index_t:])
@@ -149,5 +150,5 @@ class LightGBM(object):
 
 if __name__ == '__main__':
     L = LightGBM()
-    # L.train_model()
-    L.evaluate_model()
+    L.train_model()
+    # L.evaluate_model()