Update app.py

app.py

@@ -115,7 +115,7 @@ def train_model(X,labels):
         return
     train_data = lgb.Dataset(X, label=labels)
     num_round = 10
-    param = {'num_leaves':30, 'objective': 'binary', 'metric' : 'auc', 'lambda_l2': 0.1, 'max_depth':8}
+    param = {'objective': 'binary', 'metric' : 'auc', 'lambda_l2': 0.5}
     bst = lgb.train(param, train_data, num_round)
     in_sample_preds = bst.predict(X)
     in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]
@@ -136,6 +136,7 @@ def train():
     bst.save_model(filename)
     preds = rank_candidates(bst)
     st.session_state.preds = preds
+    st.balloons()