Skip to content

Commit 2a4a6dc

Browse files
committed
Throw exception if all scale factors are 0
1 parent 5a6f963 commit 2a4a6dc

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

niaarm/niaarm.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,16 @@ def __init__(self, dimension, features, transactions, alpha=0.0, beta=0.0, gamma
3636
"""
3737
self.features = features
3838
self.transactions = transactions
39+
40+
if alpha + beta + gamma + delta == 0:
41+
raise ValueError('At least one of alpha, beta, gamma or delta must be set')
42+
3943
self.alpha = alpha
4044
self.beta = beta
4145
self.gamma = gamma
4246
self.delta = delta
43-
self.logging = logging
4447

48+
self.logging = logging
4549
self.best_fitness = np.NINF
4650
self.rules = []
4751
super().__init__(dimension, 0.0, 1.0)

0 commit comments

Comments
 (0)