def test_glm_fit():
X = np.arange(-2, 2, .01)[:, np.newaxis]
idxs = range(X.shape[0])
idxs = random.sample(idxs, 200)
X = X[idxs]
Z = np.sin(X)
glm = GeneralizedLinearModel(1, 1, max_iter=10)
glm.fit(X, Z)
def test_glm_iter_fit():
X = np.arange(-2, 2, .01)[:, np.newaxis]
idxs = range(X.shape[0])
idxs = random.sample(idxs, 200)
X = X[idxs]
Z = np.sin(X)
glm = GeneralizedLinearModel(1, 1, max_iter=10)
for i, info in enumerate(glm.iter_fit(X, Z)):
if i >= 10:
break
