Fix dimension calculation in _base.py and add docstring params in _loss.py

Author: statmlben

rehline/_base.py

@@ -278,7 +278,8 @@ def _make_loss_rehline_param(loss, X, y):
         The +/- labels for class membership of each sample.
     """
 
-    n, d = X.shape
+    # n, d = X.shape
+    n = len(y)
 
     ## initialization of ReHLine params
     U=np.empty(shape=(0,0))

rehline/_loss.py

@@ -64,6 +64,8 @@ def __init__(self, relu_coef, relu_intercept,
     def __call__(self, x):
         """Evaluate ReHLoss given a data matrix
 
+        Parameters
+        ----------
         x: {array-like} of shape (n_samples, )
             Training vector, where `n_samples` is the number of samples
         """