Method __init__

data_loader.py:12–36

Args: filename: a list of pickle files. transform: transform applied to the feature data. target_transform: transform applied to the label data. target: target label encoding approach. Notes:

(self, filename, transform=None, target_transform=None, target="hot")

class heter_data(data.Dataset):
    def __init__(self, filename, transform=None, target_transform=None, target="hot"):
        """
            Args:
                filename: a list of pickle files.
                transform: transform applied to the feature data.
                target_transform: transform applied to the label data.
                target: target label encoding approach.
            Notes:
                Input data with key values: "feature", "label" (already one-hot encoded), "user"(optional).
                Feature with shape [seq_len, 8*2,  interval_len] ([12, 16, 7])
                For a single file we have:
        """
        self.transform = transform
        self.target_transform = target_transform
        self.data = []
        self.targets = []

        for file in filename:
            data = utils.load_pickle(file)
            self.data.extend(data["feature"])
            if target == "hot":  # is one-hot required?
                self.targets.extend(data["label"])
            else:   # Train classifier, don't need one-hot encoding
                self.targets.extend(np.argmax(data["label"], axis=1))
        self.targets = np.array(self.targets)

    def __getitem__(self, index):
        img, target = self.data[index], self.targets[index]