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

Before Change

    build_dict(label_data_path, label_dict_path)
    // load dict
    word_ids_dict, ids_word_dict = load_dict(word_dict_path), load_reverse_dict(word_dict_path)
    label_ids_dict, ids_label_dict = load_dict(label_dict_path), load_reverse_dict(label_dict_path)
    // Read data to index
    word_ids = data_reader(word_data_path, word_ids_dict)
    label_ids = data_reader(label_data_path, label_ids_dict)

After Change

    print("y_test.shape", y_test.shape)
    logger.info("Data loaded.")

    logger.info("Training BILSTM_CRF model...")
    model = create_model(word_ids_dict, label_ids_dict,
                         embedding_dim, rnn_hidden_dim)
    // fit
    model.fit(X_train, y_train,

Instances