modify unit test (bc3a1e14) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/example_DTNN.mat

−2.69 KiB (10.1 KiB)

File changed.

No diff preview for this file type.

+3 −3

Original line number	Diff line number	Diff line
		@@ -675,9 +675,9 @@ class TestOverfit(test_util.TensorFlowTestCase):
		w = np.ones_like(y)
		dataset = dc.data.DiskDataset.from_numpy(X, y, w, ids=None)
		regression_metric = dc.metrics.Metric(
		dc.metrics.r2_score, task_averager=np.mean)
		dc.metrics.pearson_r2_score, task_averager=np.mean)
		n_tasks = y.shape[1]
		max_n_atoms = list(dataset.get_data_shape())
		max_n_atoms = list(dataset.get_data_shape())[0]
		batch_size = 10

		graph_model = dc.nn.SequentialDTNNGraph(max_n_atoms=max_n_atoms)
		@@ -704,7 +704,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
		# Eval model on train
		scores = model.evaluate(dataset, [regression_metric])

		assert scores[regression_metric.name] < .2
		assert scores[regression_metric.name] > .9

		def test_siamese_singletask_classification_overfit(self):
		"""Test siamese singletask model overfits tiny data."""