fix test failure (d7ff882e) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_overfit.py

+2 −2

Original line number	Original line	Diff line number	Diff line
	@@ -685,7 +685,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
	graph_model.add(dc.nn.DTNNStep(n_embedding=20))		graph_model.add(dc.nn.DTNNStep(n_embedding=20))
	graph_model.add(dc.nn.DTNNGather(n_embedding=20))		graph_model.add(dc.nn.DTNNGather(n_embedding=20))
	n_feat = 20		n_feat = 20
	model = dc.models.DTNNGraphRegressor(		model = dc.models.MultitaskGraphRegressor(
	graph_model,		graph_model,
	n_tasks,		n_tasks,
	n_feat,		n_feat,
	@@ -736,7 +736,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
	n_tasks,		n_tasks,
	n_feat,		n_feat,
	batch_size=batch_size,		batch_size=batch_size,
	learning_rate=0.01,		learning_rate=0.001,
	learning_rate_decay_time=1000,		learning_rate_decay_time=1000,
	optimizer_type="adam",		optimizer_type="adam",
	beta1=.9,		beta1=.9,