Commit f379159a authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #246 from peastman/bug

Bug fix
parents 220647e1 93a5f59b
Loading
Loading
Loading
Loading
+8 −6
Original line number Diff line number Diff line
@@ -177,11 +177,13 @@ class Dataset(object):
    n = 0
    for X, y, _, _ in self.itersamples():
        n += 1
        if X_stats:
            dx = X-X_means
        dy = y-y_means
            X_means += dx/n
        y_means += dy/n
            X_m2 += dx*(X-X_means)
        if y_stats:
            dy = y-y_means
            y_means += dy/n
            y_m2 += dy*(y-y_means)
    if n < 2:
        X_stds = 0.0