Skip to content

Commit c704fe3

Browse files
authored
Fix score value
1 parent 115f97a commit c704fe3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

mnist/xgboost3.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
"num_boost_round" : [5, 10, 15],
3131
}
3232
for param in ParameterGrid(param_grid):
33+
train_avg_score = 0
34+
test_avg_score = 0
3335
print(param)
3436
for train_index, test_index in KFold(n_splits=k).split(X):
3537
train_X = X.loc[train_index,:]
@@ -63,4 +65,4 @@
6365
bst = xgb.train(param, dtrain)
6466

6567
with open("m003.model","wb") as f:
66-
pickle.dump(bst, f)
68+
pickle.dump(bst, f)

0 commit comments

Comments
 (0)