Commit 94404e26 authored by miaecle's avatar miaecle
Browse files

fix jenkins

parent e98ccc8b

deepchem/molnet/run_benchmark.py

+0 −1
Original line number Diff line number Diff line
@@ -71,7 +71,6 @@ def run_benchmark(datasets, 
      if metric == None:

        metric = [

            deepchem.metrics.Metric(deepchem.metrics.roc_auc_score, np.mean),

            deepchem.metrics.Metric(deepchem.metrics.prc_auc_score, np.mean)

        ]

    elif dataset in [

        'bace_r', 'chembl', 'clearance', 'delaney', 'hopv', 'kaggle', 'lipo',

devtools/jenkins/compare_results.py

+4 −4
Original line number Diff line number Diff line
@@ -38,11 +38,11 @@ def parse_test_results(test_results): 
  for line in test_results:

    vars = line.split(',')

    retval.append({

        "split": BENCHMARK_TO_DESIRED_KEY_MAP[vars[2]],

        "data_set": vars[1],

        "model": BENCHMARK_TO_DESIRED_KEY_MAP[vars[5]],

        "split": BENCHMARK_TO_DESIRED_KEY_MAP[vars[1]],

        "data_set": vars[0],

        "model": BENCHMARK_TO_DESIRED_KEY_MAP[vars[3]],

        "train_score": float(vars[6]),

        "test_score": float(vars[9])

        "test_score": float(vars[8])

    })

  return retval

CRA Git | Maintained and supported by SUSTech CRA and CCSE