Python QuestionAnsweringExplainer.visualize Examples

Example #1

File: test_question_answering_explainer.py Project: zjcanjux/transformers-interpret

def test_question_answering_visualize():
    qa_explainer = QuestionAnsweringExplainer(DISTILBERT_QA_MODEL,
                                              DISTILBERT_QA_TOKENIZER)
    explainer_question = "what is his name ?"
    explainer_text = "his name is Bob"
    qa_explainer(explainer_question, explainer_text)
    qa_explainer.visualize()

Example #2

def test_question_answering_visualize_save_append_html_file_ending():
    qa_explainer = QuestionAnsweringExplainer(DISTILBERT_QA_MODEL,
                                              DISTILBERT_QA_TOKENIZER)
    explainer_question = "what is his name ?"
    explainer_text = "his name is Bob"
    qa_explainer(explainer_question, explainer_text)

    html_filename = "./test/qa_test"
    qa_explainer.visualize(html_filename)
    assert os.path.exists(html_filename + ".html")
    os.remove(html_filename + ".html")