7cdd969aeb3af048b1573f4dac1d0a672ef4576b,anago/train.py,,main,#,17

Before Change

    lstm_dim = 100
    batch_size = 64

    max_features = len(word_to_id)
    nb_chunk_tags = len(entity_to_id)

    X_train = sequence.pad_sequences(train_data["X"], maxlen=maxlen, padding="post")
    X_test = sequence.pad_sequences(test_data["X"], maxlen=maxlen, padding="post")

After Change

    if args.save_path:
        print("Saving model to {}.".format(args.save_path))
        model.save(args.save_path)


if __name__ == "__main__":
    main()

Instances