code: def return_train_dataset(self):
          
          X, y = self.return_main_dataset()
  
          if self.test_dataset['method'] == 'split_from_main':
              X, X_test, y, y_test = train_test_split(
                  X,
                  y,
                  test_size=self.test_dataset['split_ratio'],
                  random_state=self.test_dataset['split_seed'],
                  stratify=y
              )
  
          return X, y
docs: Returns train data set
  
          Returns:
              X (numpy.ndarray): Features
  
              y (numpy.ndarray): Labels