Better reload test (663077ef) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_reload.py

+3 −4

Original line number	Diff line number	Diff line
		@@ -75,7 +75,7 @@ class TestReload(unittest.TestCase):

		# Fit trained model
		model.fit(dataset)
		scores_preload = model.evaluate(dataset, [classification_metric])
		scores_preload = model.evaluate(dataset, [classification_metric])['accuracy_score']
		model.save()

		# Load trained model
		@@ -84,6 +84,5 @@ class TestReload(unittest.TestCase):
		reloaded_model.reload()

		# Eval model on train
		scores_reload = reloaded_model.evaluate(dataset, [classification_metric])
		assert np.all(
		np.isclose(scores_preload.values(), scores_reload.values(), 0.01))
		scores_reload = reloaded_model.evaluate(dataset, [classification_metric])['accuracy_score']
		assert np.isclose(scores_preload, scores_reload, 0.01)