langtest.transform.utils.DataRetriever#

class DataRetriever#

Bases: object

DataRetriever class to load and filter data from csv files

cases#

Dataframe containing case data

Type:

pd.DataFrame

questions#

Dataframe containing question data

Type:

pd.DataFrame

sections#

Dataframe containing section data

Type:

pd.DataFrame

criteria#

Dataframe containing criteria

Type:

pd.DataFrame

filter(df

pd.DataFrame, **conditions) -> pd.DataFrame: Filter the dataframe based on conditions

get_case_data(case_id) → Tuple[pd.Series, pd.DataFrame]#

Get case data based on case_id

get_question_data(case_id, question_id) → Tuple[pd.Series, pd.DataFrame]#

Get question data based on case_id and question_id

get_criteria(case_id, question_id, section_id=None) → List[str]#

Get criteria based on case_id, question_id and section_id

get_criteria_scores(case_id, question_id, section_id=None) → List[float]#

Get criteria scores based on case_id, question_id and section_id

load_csv(filename) → pd.DataFrame#

Load csv file from github

__init__()#

Methods

__init__()
dataset_info()	Get dataset information
filter(df, **conditions)
get_case_data(case_id)
get_cases()
get_criteria(case_id, question_id[, section_id])
get_criteria_scores(case_id, question_id[, ...])
get_question_data(case_id, question_id)
load_csv(filename)

dataset_info() → DataFrame#

Get dataset information

Returns:

Dataset information

Return type:

pd.DataFrame