p1b2_ljeabmreosn_mlp.py 文件源码

def train_model(model):
    opt = 'rmsprop'
    model_checkpoint = ModelCheckpoint(
        filepath=BEST_MODEL_PATH, 
        monitor='val_acc', 
        verbose=0, 
        save_best_only=True,  
        mode='auto'
    )
    overfitting_stopper = EarlyStopping(
        monitor='val_acc', 
        min_delta=0, 
        patience=5, 
        verbose=1, 
        mode='auto'
    )
    model.compile(loss='categorical_crossentropy', optimizer=opt, metrics=['accuracy'])
    model_history = model.fit(X_train, y_train,
        batch_size       = 64,
        nb_epoch         = 50,
        #shuffle          = True,
        validation_split = 0.2,
        #verbose          = 2
        callbacks        = [overfitting_stopper, model_checkpoint]
    )