Update feedforward.py (#15)

* Update feedforward.py

* Update models/feedforward.py

Co-authored-by: Eric Ma <[email protected]>

Author: abdullahahmed1995

models/feedforward.py

@@ -52,6 +52,7 @@ def fit(self, X, y, n=200000, batch_size=10):
         Train the Bayesian NN model.
         """
         num_samples, self.num_pred = X.shape
+        _, self.num_out = y.shape
 
         if self.cached_model is None:
             self.cached_model = self.create_model()
@@ -66,4 +67,3 @@ def fit(self, X, y, n=200000, batch_size=10):
         return self
 
 
-