:rotating_light: fix lint (3c1ba819) · Commits · 钟慕尧 / deepchem

deepchem/hyper/tests/test_gaussian_hyperparam_opt.py

+1 −7

Original line number	Diff line number	Diff line
		@@ -42,11 +42,7 @@ class TestGaussianHyperparamOpt(unittest.TestCase):
		metric = dc.metrics.Metric(dc.metrics.pearson_r2_score)

		best_model, best_hyperparams, all_results = optimizer.hyperparam_search(
		params_dict,
		self.train_dataset,
		self.valid_dataset,
		metric,
		max_iter=2)
		params_dict, self.train_dataset, self.valid_dataset, metric, max_iter=2)

		valid_score = best_model.evaluate(self.valid_dataset, [metric],
		transformers)
		@@ -111,7 +107,6 @@ class TestGaussianHyperparamOpt(unittest.TestCase):
		learning_rate=0.003, **params))

		params_dict = {"batch_size": 10}
		transformers = []
		metric = dc.metrics.Metric(
		dc.metrics.mean_squared_error, task_averager=np.mean)

		@@ -149,7 +144,6 @@ class TestGaussianHyperparamOpt(unittest.TestCase):
		params_dict = {"learning_rate": 0.003, "batch_size": 10}
		# These are per-example multiplier
		search_range = {"learning_rate": 10, "batch_size": 4}
		transformers = []
		metric = dc.metrics.Metric(
		dc.metrics.mean_squared_error, task_averager=np.mean)