File size: 545 Bytes
81d4aee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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)))
|