make Model an sklearn BaseEstimator (abd6c41b) · Commits · 钟慕尧 / deepchem

deepchem/models/models.py

+3 −2

Original line number	Diff line number	Diff line
		@@ -16,6 +16,7 @@ import joblib
		import os
		import tempfile
		import sklearn
		from sklearn.base import BaseEstimator

		from deepchem.data import Dataset, pad_features
		from deepchem.trans import undo_transforms
		@@ -25,7 +26,7 @@ from deepchem.utils.save import log
		from deepchem.utils.evaluate import Evaluator


		class Model(object):
		class Model(BaseEstimator):
		"""
		Abstract base class for different ML models.
		"""