From b763f8964f60a12df64f373546bf3d884e26ea8b Mon Sep 17 00:00:00 2001 From: fchollet Date: Tue, 12 May 2015 13:07:50 -0700 Subject: [PATCH] Improve Otto example --- examples/kaggle_otto_nn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/kaggle_otto_nn.py b/examples/kaggle_otto_nn.py index f37c5e25f..face3333a 100644 --- a/examples/kaggle_otto_nn.py +++ b/examples/kaggle_otto_nn.py @@ -31,6 +31,8 @@ from sklearn.preprocessing import StandardScaler - with smaller layers, largers layers - with more layers, less layers - with different optimizers (SGD+momentum+decay is probably better than Adam!) + + Get the data from Kaggle: https://www.kaggle.com/c/otto-group-product-classification-challenge/data ''' np.random.seed(1337) # for reproducibility @@ -113,7 +115,7 @@ model.compile(loss='categorical_crossentropy', optimizer="adam") print("Training model...") -model.fit(X, y, nb_epoch=20, batch_size=16, validation_split=0.15) +model.fit(X, y, nb_epoch=20, batch_size=128, validation_split=0.15) print("Generating submission...")