Fix for HyperparamOpt with no fit_transformers (47e0c8f5) · Commits · 钟慕尧 / deepchem

deepchem/hyperparameters/init.py

+5 −0

Original line number	Diff line number	Diff line
		@@ -60,9 +60,14 @@ class HyperparamOpt(object):
		model = self.model_class(self.task_types, model_params, model_dir,
		verbosity=self.verbosity)
		else:
		if not self.fit_transformers:
		model = self.model_class(self.task_types, model_params,
		fit_transformers=self.fit_transformers,
		verbosity=self.verbosity)
		else:
		model = self.model_class(self.task_types, model_params,
		verbosity=self.verbosity)

		model.fit(train_dataset)
		model.save(model_dir)