bugfix: 修复计算auc ks时概率错误问题

model/model_lr.py

@@ -23,11 +23,11 @@ class ModelLr(ModelBase):
         test_data = data.test_data
         self.lr.fit(train_data.get_Xdata(), train_data.get_Ydata())
 
-        train_prob = self.lr.predict(train_data.get_Xdata())
+        train_prob = self.lr.predict_proba(train_data.get_Xdata())[:, 1]
         train_auc = AUC(train_prob, train_data.get_Ydata())
         train_ks = KS(train_prob, train_data.get_Ydata())
 
-        test_prob = self.lr.predict(test_data.get_Xdata())
+        test_prob = self.lr.predict_proba(test_data.get_Xdata())[:, 1]
         test_auc = AUC(test_prob, test_data.get_Ydata())
         test_ks = KS(test_prob, test_data.get_Ydata())