def test_multinomial():
"""Test all multinomial functionality"""
glm = GLM(distr='multinomial', reg_lambda=np.array([0.0, 0.1, 0.2]), tol=1e-10)
X = np.array([[-1, -2, -3], [4, 5, 6]])
y = np.array([1, 0])
# test gradient
beta = np.zeros([4, 2])
grad_beta0, grad_beta = glm.grad_L2loss(beta[0], beta[1:], 0, X, y)
assert grad_beta0[0] != grad_beta0[1]
glm.fit(X, y)
y_pred = glm.predict(X)
assert_equal(y_pred.shape, (3, X.shape[0], 2)) # n_lambdas x n_samples x n_classes
# pick one as yhat
yhat = y_pred[0]
# uniform prediction
ynull = np.ones(yhat.shape) / yhat.shape[1]
# pseudo_R2 should be greater than 0
assert_true(glm.pseudo_R2(y, yhat, ynull) > 0.)
glm.deviance(y, yhat)
assert_equal(len(glm.simulate(glm.fit_[0]['beta0'],
glm.fit_[0]['beta'],
X)),
X.shape[0])
# these should raise an exception
try:
glm.pseudo_R2(y, y, y)
assert False
except Exception:
assert True
try:
glm.deviance(y, y)
assert False
except Exception:
assert True
def test_glmnet():
"""Test glmnet."""
scaler = StandardScaler()
n_samples, n_features = 10000, 100
density = 0.1
n_lambda = 10
# coefficients
beta0 = np.random.rand()
beta = sps.rand(n_features, 1, density=density).toarray()
distrs = ['poisson', 'poissonexp', 'normal', 'binomial']
for distr in distrs:
# FIXME: why do we need such this learning rate for 'poissonexp'?
learning_rate = 1e-5 if distr == 'poissonexp' else 1e-4
glm = GLM(distr, learning_rate=learning_rate)
assert_true(repr(glm))
np.random.seed(glm.random_state)
X_train = np.random.normal(0.0, 1.0, [n_samples, n_features])
y_train = glm.simulate(beta0, beta, X_train)
X_train = scaler.fit_transform(X_train)
glm.fit(X_train, y_train)
beta_ = glm.fit_[-2]['beta'][:]
assert_allclose(beta[:], beta_, atol=0.1) # check fit
density_ = np.sum(beta_ > 0.1) / float(n_features)
assert_allclose(density_, density, atol=0.05) # check density
y_pred = glm.predict(scaler.transform(X_train))
assert_equal(y_pred.shape, (n_lambda, X_train.shape[0]))
# checks for slicing.
glm = glm[:3]
glm_copy = glm.copy()
assert_true(glm_copy is not glm)
assert_equal(len(glm.reg_lambda), 3)
y_pred = glm[:2].predict(scaler.transform(X_train))
assert_equal(y_pred.shape, (2, X_train.shape[0]))
y_pred = glm[2].predict(scaler.transform(X_train))
assert_equal(y_pred.shape, (X_train.shape[0], ))
assert_raises(IndexError, glm.__getitem__, [2])
glm.deviance(y_train, y_pred)
# don't allow slicing if model has not been fit yet.
glm = GLM(distr='poisson')
assert_raises(ValueError, glm.__getitem__, 2)
# test fit_predict
glm.fit_predict(X_train, y_train)
assert_raises(ValueError, glm.fit_predict, X_train[None, ...], y_train)
