Merge branch 'master' of github.com:jeffrey-hokanson/PSDR

jeffrey-hokanson · Oct 10, 2019 · f0c45f6 · f0c45f6
2 parents 8c602bb + 07faf5f
commit f0c45f6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
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

diff --git a/psdr/polyridge.py b/psdr/polyridge.py
@@ -406,7 +406,6 @@ def _fit_affine(self, X, fX):
 		U /= np.linalg.norm(U)
 		return U	
 
-
 	def _fit_coef(self, X, fX, U):
 		r""" Returns the linear coefficients
 		"""