def test_poisson_fit(fit_intercept): X, y = make_counts(n_samples=100, chunks=500) pr = PoissonRegression(fit_intercept=fit_intercept) pr.fit(X, y) pr.predict(X) pr.get_deviance(X, y) if fit_intercept: assert pr.intercept_ is not None
def xy_counts(): """X, y pair for predicting counts""" X, y = make_counts(n_samples=100, n_features=5, chunks=10) return X, y