Ejemplo n.º 1
0
def main():
    config = Config()
    word_dict = geniaDataset()
    model = AttentionNestedNERModel(
        config, word_dict).cuda() if config.cuda else AttentionNestedNERModel(
            config, word_dict)

    config.test_data, config.test_str, config.test_label = data_prepare(
        config, config.get_test_path(), word_dict)
    del word_dict
    start_test(config, model)