test reload (bb904c1a) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_reload.py

+4 −2

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

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

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

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