Commit cda5a592 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent c872eaf5
Loading
Loading
Loading
Loading
+3 −8
Original line number Diff line number Diff line
@@ -11,13 +11,8 @@ train, valid, test= delaney_datasets
# Fit models
regression_metric = dc.metrics.Metric(dc.metrics.pearson_r2_score)


# TODO(rbharath): I don't like this awkward string/class divide. Maybe clean up?
optimizer = dc.hyper.GaussianProcessHyperparamOpt('tf_regression')
best_hyper_params, best_performance = optimizer.hyperparam_search(
  dc.molnet.preset_hyper_parameters.hps['tf_regression'],
  train,
  valid,
  transformers,
  [regression_metric]
)
    dc.molnet.preset_hyper_parameters.hps['tf_regression'], train, valid,
    transformers, [regression_metric])