03ef391a07e08eefce7299f3a934b49505c5be16,examples/pdbbind/pdbbind_datasets.py,,featurize_pdbbind,#Any#Any#Any#,70

Before Change

      print("%s is missing!" % pdb_subdir)
      missing_pdbs.append(pdb_subdir)
      continue
    computed_feature = compute_pdbbind_features(featurizer, pdb_subdir,
                                                pdb_code)
    if feature_len is None:
      feature_len = len(computed_feature)
    if len(computed_feature) != feature_len:
      print("Featurization failed for %s!" % pdb_code)
      continue
    y_inds.append(ind)
    features.append(computed_feature)
  time2 = time.time()

After Change

  features = []
  y_inds = []
  time1 = time.time()
  p = Pool(multiprocessing.cpu_count())
  args = []
  for ind, pdb_code in enumerate(ids):
    args.append((ind, pdb_code, pdbbind_dir, featurizer))
  results = p.map(compute_single_pdbbind_feature, args)

Instances