data.trial_data

class pytrial.data.trial_data.TrialDatasetBase(data, criteria_column='criteria')

The basic trial datasets loader.

Parameters

• data (pd.DataFrame) – Contain the trial document in tabular format.

• criteria_column (str) – The column name of eligibility criteria in the dataframe.

get_ec_sentence_embedding()

Process the eligibility criteria of each trial, get the criterion-level emebddings stored in dict.

Parameters

criteria_column (str) – The column name of eligibility criteria in the dataframe.

class pytrial.data.trial_data.TrialOutcomeDatasetBase(data, columns=None)

Basic trial outcome datasets loader.

Parameters

data (pd.DataFrame) – Contain the trial document in tabular format.