Skip to content

Commit e9ba390

Browse files
committed
Kaggle-Otto: it's the validation set, not test set.
1 parent b96a500 commit e9ba390

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/Kaggle-Otto/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def nnet():
4848

4949
train_data_x, train_data_y = load_train_data()
5050

51-
train_data_x, test_data_x, train_data_y, test_data_y = train_test_split(train_data_x, train_data_y, train_size=0.85)
51+
train_data_x, valid_data_x, train_data_y, valid_data_y = train_test_split(train_data_x, train_data_y, train_size=0.85)
5252
model = nnet()
5353

5454
criterion = bb8.ClassNLLCriterion()
@@ -63,5 +63,5 @@ def nnet():
6363
train(train_data_x, train_data_y, model, optimiser, criterion, epoch, 100, 'stats')
6464

6565
model.evaluate()
66-
validate(test_data_x, test_data_y, model, epoch, 100)
66+
validate(valid_data_x, valid_data_y, model, epoch, 100)
6767

0 commit comments

Comments
 (0)