def prep_transform(data): data, y, gold = data data = dt.load_prepped(data) data = dt.prepped_to_tensor(data) data.y = th.tensor(y).unsqueeze(0).float() data.gold = th.tensor(gold).unsqueeze(0).float() return data
def unprep_transform(data): data, y, gold = data name = data.name data = dt.prep_data(data) data = dt.prepped_to_tensor(data) data.name = name data.y = th.tensor(y).unsqueeze(0).float() data.gold = th.tensor(gold).unsqueeze(0).float() return data
def unprep_transform_wo_labels(data): data = dt.prep_data(data) data = dt.prepped_to_tensor(data) return data
def prep_transform_wo_labels(data): data = dt.load_prepped(data) data = dt.prepped_to_tensor(data) return data