code: def _load_dataset(self, images, labels, emotion_index_map):
          
          train_images, test_images, train_labels, test_labels = train_test_split(images, labels, test_size=self.validation_split, random_state=42, stratify=labels)
          dataset = Dataset(train_images, test_images, train_labels, test_labels, emotion_index_map, self.time_delay)
          return dataset
docs: Loads Dataset object with images, labels, and other data.
  
          :param images: numpy array of image data
          :param labels: numpy array of one-hot vector labels
          :param emotion_index_map: map linking string/integer emotion class to integer index used in labels vectors
  
          :return: Dataset object containing image and label data.