Made test more tolerant (d0dfee5b) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_kerasmodel.py

+2 −5

Original line number	Diff line number	Diff line
		@@ -27,10 +27,7 @@ class TestKerasModel(unittest.TestCase):
		learning_rate=0.005)
		model.fit(dataset, nb_epoch=1000)
		prediction = np.squeeze(model.predict_on_batch(X))
		print(y)
		print(prediction)
		print(np.isclose(prediction, y.flatten(), atol=0.4))
		assert np.all(np.isclose(prediction, y.flatten(), atol=0.4))
		assert np.array_equal(y, np.round(prediction))
		metric = dc.metrics.Metric(dc.metrics.roc_auc_score)
		scores = model.evaluate(dataset, [metric])
		assert scores[metric.name] > 0.9
		@@ -55,7 +52,7 @@ class TestKerasModel(unittest.TestCase):
		keras_model, dc.models.losses.BinaryCrossEntropy(), learning_rate=0.005)
		model.fit(dataset, nb_epoch=1000)
		prediction = np.squeeze(model.predict_on_batch(X))
		assert np.all(np.isclose(prediction, y.flatten(), atol=0.4))
		assert np.array_equal(y, np.round(prediction))
		metric = dc.metrics.Metric(dc.metrics.roc_auc_score)
		generator = model.default_generator(dataset)
		scores = model.evaluate_generator(generator, [metric])