Potential bugfix (8216fd80) · Commits · 钟慕尧 / deepchem

deepchem/models/multitask.py

+4 −1

Original line number	Diff line number	Diff line
		@@ -67,7 +67,10 @@ class SingletaskToMultitask(Model):
		for ind, task in enumerate(self.tasks):
		task_type = self.task_types[task]
		if task_type == "classification":
		y_pred[:, ind] = self.models[task].predict_on_batch(X)[:, 0]
		################# DEBUG
		#y_pred[:, ind] = self.models[task].predict_on_batch(X)[:, 0]
		################# DEBUG
		y_pred[:, ind] = self.models[task].predict_on_batch(X)
		elif task_type == "regression":
		y_pred[:, ind] = self.models[task].predict_on_batch(X)
		else: