Added predict to BaseFunction

Added predict to the BaseFunction class to better enable compatability with scikit learn. Based on a suggestion from luk-f-a in #28.
jeffrey-hokanson · Oct 2, 2019 · 07faf5f · 07faf5f
1 parent b38c31e
commit 07faf5f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/psdr/function.py b/psdr/function.py
@@ -35,6 +35,11 @@ def __call__(self, X, return_grad = False, **kwargs):
 		else:
 			return self.eval(X, **kwargs)
 
+	def predict(self, X):
+		r""" Alias of __call__ to match scikit learn API
+		"""
+		return self.__call__(X)
+
 
 class Function(BaseFunction):
 	r"""Wrapper around function specifying the domain