4239066b2ac566c8f805fe5bade39815f416de89,examples/adme/run_benchmarks.py,,,#,136

Before Change

      best_val_score = latest_val_score
      train_score = latest_train_score
  print((MODEL, SPLIT, DATASET, train_score, best_val_score))
elif MODEL == "PDNN":
  model.fit(train, nb_epoch=25)
  train_score = model.evaluate(train, [metric],
                               transformers)["mean-pearson_r2_score"]
  val_score = model.evaluate(val, [metric],
                             transformers)["mean-pearson_r2_score"]
  print((MODEL, SPLIT, DATASET, train_score, val_score))
  // we cant re-open the closed session...
else:
  model.fit(train)
  train_score = model.evaluate(train, [metric],

After Change

//======================================================================
// Run Benchmarks {GC-DNN, SVR, RF}

print("About to retrieve datasets")
retrieve_datasets()

MODEL = "GraphConv"
SPLIT = "scaffold"

Instances