Pin Random Seeds for Generator Evaluator Test (7164be07) · Commits · 钟慕尧 / deepchem

deepchem/utils/test/test_generator_evaluator.py

+12 −8

Original line number	Diff line number	Diff line
		@@ -5,12 +5,17 @@ from unittest import TestCase
		import deepchem as dc
		from deepchem.data import NumpyDataset
		from deepchem.data.datasets import Databag
		from deepchem.models.tensorgraph.layers import Dense, ReduceMean, SoftMax, SoftMaxCrossEntropy, L2Loss
		from deepchem.models.tensorgraph.layers import Dense, ReduceMean, SoftMax, \
		SoftMaxCrossEntropy, L2Loss
		from deepchem.models.tensorgraph.layers import Feature, Label
		from deepchem.models.tensorgraph.layers import ReduceSquareDifference

		RANDOM_SEED = 42


		def set_random_seeds():
		np.random.seed(42)
		np.random.seed(RANDOM_SEED)


		class TestGeneratorEvaluator(TestCase):

		@@ -50,7 +55,7 @@ class TestGeneratorEvaluator(TestCase):

		total_loss = ReduceMean(in_layers=entropies)

		tg = dc.models.TensorGraph(learning_rate=0.1)
		tg = dc.models.TensorGraph(learning_rate=0.1, random_seed=RANDOM_SEED)
		for output in outputs:
		tg.add_output(output)
		tg.set_loss(total_loss)
		@@ -102,7 +107,7 @@ class TestGeneratorEvaluator(TestCase):

		total_loss = ReduceMean(in_layers=entropies)

		tg = dc.models.TensorGraph(learning_rate=0.1)
		tg = dc.models.TensorGraph(learning_rate=0.1, random_seed=RANDOM_SEED)
		for output in outputs:
		tg.add_output(output)
		tg.set_loss(total_loss)
		@@ -120,10 +125,8 @@ class TestGeneratorEvaluator(TestCase):

		@with_setup(set_random_seeds)
		def test_compute_model_performance_multitask_regressor(self):
		random_seed = 42
		n_data_points = 20
		n_features = 2
		np.random.seed(seed=random_seed)

		X = np.random.rand(n_data_points, n_features)
		y1 = np.expand_dims(np.array([0.5 for x in range(n_data_points)]), axis=-1)
		@@ -154,7 +157,7 @@ class TestGeneratorEvaluator(TestCase):
		tg = dc.models.TensorGraph(
		mode="regression",
		batch_size=20,
		random_seed=random_seed,
		random_seed=RANDOM_SEED,
		learning_rate=0.1)
		for output in outputs:
		tg.add_output(output)
		@@ -202,7 +205,8 @@ class TestGeneratorEvaluator(TestCase):

		total_loss = ReduceMean(in_layers=losses)

		tg = dc.models.TensorGraph(mode="regression", learning_rate=0.1)
		tg = dc.models.TensorGraph(
		mode="regression", learning_rate=0.1, random_seed=RANDOM_SEED)
		for output in outputs:
		tg.add_output(output)
		tg.set_loss(total_loss)

Admin message