skanderovitch commited on
Commit
b8721b8
·
verified ·
1 Parent(s): f6230c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def train_model(X,labels):
115
  return
116
  train_data = lgb.Dataset(X, label=labels)
117
  num_round = 10
118
- param = {'num_leaves':50, 'objective': 'binary', 'metric' : 'auc', 'lambda_l2': 0.1, 'max_depth':5}
119
  bst = lgb.train(param, train_data, num_round)
120
  in_sample_preds = bst.predict(X)
121
  in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]
 
115
  return
116
  train_data = lgb.Dataset(X, label=labels)
117
  num_round = 10
118
+ param = {'num_leaves':30, 'objective': 'binary', 'metric' : 'auc', 'lambda_l2': 0.1, 'max_depth':8}
119
  bst = lgb.train(param, train_data, num_round)
120
  in_sample_preds = bst.predict(X)
121
  in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]