skanderovitch commited on
Commit
cac7657
·
verified ·
1 Parent(s): 5c562da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def train_model(X,labels):
79
  return
80
  train_data = lgb.Dataset(X, label=labels)
81
  num_round = 10
82
- param = {'num_leaves':10, 'objective': 'binary', 'metric' : 'auc'}
83
  bst = lgb.train(param, train_data, num_round)
84
  in_sample_preds = bst.predict(X)
85
  in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]
 
79
  return
80
  train_data = lgb.Dataset(X, label=labels)
81
  num_round = 10
82
+ param = {'num_leaves':10, 'objective': 'binary', 'metric' : 'binary'}
83
  bst = lgb.train(param, train_data, num_round)
84
  in_sample_preds = bst.predict(X)
85
  in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]