Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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' : '
|
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]
|