import Keras_model_deep_noLSTM as Akar from vqa import sent_load_data # Parameters batch_size = 200 nb_epoch = 20 verbose = 1 validation_split = 0.0 shuffle = True show_accuracy = True MODEL_ROOT = '../models/elephas/' MODEL = 'sent_overfit_noLSTM_batch_{}'.format(batch_size) MODEL_NAME = MODEL_ROOT + MODEL + ".h5" print 'Loading data...' (X_train,Y_train) = sent_load_data()[0] print 'Building model...' model = Akar.keras_model(1) #print 'Defining callbacks...' #earlystopping = EarlyStopping(monitor='val_loss', patience=2, verbose=0) print 'Start training...' for epoch in [20,40,60,80,100,120,140,160,180,200]: model.fit( X_train, Y_train, batch_size=batch_size, nb_epoch=nb_epoch, verbose=verbose, callbacks=[],validation_split=validation_split, shuffle=shuffle,show_accuracy=show_accuracy) model.save_weights(MODEL_ROOT+MODEL+"_{}.h5".format(epoch))