Beispiel #1
0
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
Beispiel #2
0
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
Beispiel #3
0
def unprep_transform_wo_labels(data):

    data = dt.prep_data(data)
    data = dt.prepped_to_tensor(data)
    return data
Beispiel #4
0
def prep_transform_wo_labels(data):

    data = dt.load_prepped(data)
    data = dt.prepped_to_tensor(data)
    return data