Commit 891e17bc authored by peastman's avatar peastman
Browse files

Added a test case

parent 5fdefa5a
Loading
Loading
Loading
Loading
+11 −0
Original line number Diff line number Diff line
@@ -679,6 +679,17 @@ def test_multitask_regressor_uncertainty():
  assert noise < np.mean(std) < 1.0


def test_multitask_regressor_delaney_uncertainty():
  """Test computing uncertainty on a larger dataset."""
  tasks, datasets, transformers = dc.molnet.load_delaney('ECFP')
  train_dataset, valid_dataset, test_dataset = datasets
  model = dc.models.MultitaskRegressor(len(tasks), 1024, uncertainty=True)
  model.fit(train_dataset, nb_epoch=20)
  metric = dc.metrics.Metric(dc.metrics.pearsonr)
  scores = model.evaluate(test_dataset, [metric], transformers)
  assert scores['pearsonr'] > 0.5


@pytest.mark.slow
def test_DAG_singletask_regression_overfit():
  """Test DAG regressor multitask overfits tiny data."""