Models:
Philip Bordner
philipB/
covid-ehr-benchmarks
0
Downloads: 1
[d6904d]: / app / datasets / base.py

Download this file

12 lines (8 with data), 333 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
import pickle





def load_data(dataset_type):

    # Load data

    data_path = f"datasets/{dataset_type}/processed_data/"

    x = pickle.load(open(data_path + "x.pkl", "rb"))

    y = pickle.load(open(data_path + "y.pkl", "rb"))

    x_lab_length = pickle.load(open(data_path + "visits_length.pkl", "rb"))



    return x, y, x_lab_length