From 26e51e44637c754d33fd65815479b22e80616f19 Mon Sep 17 00:00:00 2001 From: Dex Groves Date: Mon, 21 Dec 2020 10:54:06 +0000 Subject: [PATCH] Fix call to column_stack --- glmnet_python/lognet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/glmnet_python/lognet.py b/glmnet_python/lognet.py index d231ec7..81f5135 100644 --- a/glmnet_python/lognet.py +++ b/glmnet_python/lognet.py @@ -71,7 +71,7 @@ def lognet(x, is_sparse, irs, pcs, y, weights, offset, parm, raise ValueError('offset should have the same number of values as observations in binominal/multinomial call to glmnet') if nc == 1: if do[1] == 1: - offset = scipy.column_stack((offset, -offset), 1) + offset = scipy.column_stack((offset, -offset)) if do[1] > 2: raise ValueError('offset should have 1 or 2 columns in binomial call to glmnet') if (family == 'multinomial') and (do[1] != nc):