def test_validate_rel_penalties(self):
X = np.random.random(size=(50,10))
w = np.random.random(size=(10,))
y = np.dot(X, w)
enet = ElasticNet(alpha=.5)
# Invalid use:
# Passing in a rel_penalties vector that is too short.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(9,))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_penalties that matches the wrong dimenstion of X.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(50,))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_penalties containing a negative entry.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(10,))
rel_pens[5] = -1
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_panalties with no positive entries.
with self.assertRaises(ValueError):
rel_pens = np.zeros(shape=(10,))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Valid use:
# Rel_penalties has the correct dimenstion with all non-negative
# entries.
rel_pens = np.ones(shape=(10,))
rel_pens[5] = 0
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
def test_validate_rel_penalties(self):
X = np.random.random(size=(50, 10))
w = np.random.random(size=(10, ))
y = np.dot(X, w)
enet = ElasticNet(alpha=.5)
# Invalid use:
# Passing in a rel_penalties vector that is too short.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(9, ))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_penalties that matches the wrong dimenstion of X.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(50, ))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_penalties containing a negative entry.
with self.assertRaises(ValueError):
rel_pens = np.ones(shape=(10, ))
rel_pens[5] = -1
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Invalid use:
# Passing in a rel_panalties with no positive entries.
with self.assertRaises(ValueError):
rel_pens = np.zeros(shape=(10, ))
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
# Valid use:
# Rel_penalties has the correct dimenstion with all non-negative
# entries.
rel_pens = np.ones(shape=(10, ))
rel_pens[5] = 0
enet._validate_rel_penalties(X, y, rel_penalties=rel_pens)
