| from .cleaning import remove_citations, split_data, split_text, chunk_data | |
| import pandas as pd | |
| import numpy as np | |
| import json | |
| with open('utils/id2label.json', 'r') as j: | |
| id2label = json.loads(j.read()) | |
| with open('utils/label2id.json', 'r') as j: | |
| label2id = json.loads(j.read()) | |
| def find_case_by_name(df, name): | |
| return display(HTML(df[df['case_name'].str.contains(name)].iloc[:,:-1].to_html(render_links=True, escape=False))) | |
| def head_df(df): | |
| return display(HTML(df.iloc[:,:-1].head().to_html(render_links=True, escape=False))) | |