def _data_preprocessing()

in tfx_addons/feature_selection/component.py [0:0]

• 7 lines of code
• 6 McCabe index (conditional complexity)

def _data_preprocessing(np_dataset, target_feature):

  # getting the required data without any metadata
  np_dataset = [{k: v[0]
                 for k, v in example.items()} for example in np_dataset]

  # extracting `y`
  target = [i.pop(target_feature) for i in np_dataset]
  feature_keys = list(np_dataset[0].keys())
  # getting `X`
  input_data = [[i[j] for j in feature_keys] for i in np_dataset]

  return [feature_keys, target, input_data]