Commit 886f2795 authored by Joe Gomes's avatar Joe Gomes
Browse files

Fix comment in Model.predict

parent cd68a32d
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -147,7 +147,7 @@ class Model(object):
    batch_size = self.model_params["batch_size"]
    for (X_batch, y_batch, w_batch, ids_batch) in dataset.iterbatches(batch_size):

      # HACK(JG): This is a hack to perform n-fold averaging of y_pred on
      # HACK(JG): This was a hack to perform n-fold averaging of y_pred on
      # a given X_batch.  If fit_transformers exist, we will apply them to
      # X_batch 1 times and average the resulting y_pred before we undo 
      # transforms on y_pred and y.  In the future the averaging will be