Commit eb302b69 authored by miaecle's avatar miaecle
Browse files

little fix

parent 6d91f03c

deepchem/molnet/run_benchmark.py

+2 −2
Original line number Diff line number Diff line
@@ -66,14 +66,14 @@ def run_benchmark(datasets, 
    ]:

      mode = 'classification'

      if metric == None:

        metric = 'auc'

        metric = str('auc')

    elif dataset in [

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

        'nci', 'pdbbind', 'ppb', 'qm7', 'qm7b', 'qm8', 'qm9', 'sampl'

    ]:

      mode = 'regression'

      if metric == None:

        metric = 'r2'

        metric = str('r2'）

    else:

      raise ValueError('Dataset not supported')

deepchem/molnet/run_benchmark_low_data.py

+1 −1
Original line number Diff line number Diff line
@@ -62,7 +62,7 @@ def run_benchmark_low_data(datasets, 
    if dataset in ['muv', 'sider', 'tox21']:

      mode = 'classification'

      if metric == None:

        metric = 'auc'

        metric = str('auc')

    else:

      raise ValueError('Dataset not supported')

CRA Git | Maintained and supported by SUSTech CRA and CCSE