77a383e645c7b42ea025511738638e1238b971bb,pycorrector/rnn_crf/train.py,,train,#,21

Before Change

    model = rnn_crf(X_train, y_train, X_test, y_test, batch_size, word_ids_dict, label_ids_dict,
                    embedding_dim, rnn_hidden_dim, epoch)
    // predict
    y_test_pred = model.predict(word_seq).argmax(-1)
    // save predict file


    logger.info("Training has finished.")

After Change

    model.fit(X_train, y_train,
              batch_size=batch_size,
              epochs=epoch,
              validation_data=[X_test, y_test])
    // save model
    save_model(model, save_model_path)
    logger.info("model saved: %s" % save_model_path)

Instances