Skip to content

Commit 65b2c45

Browse files
author
SAAS R7 User1
committed
update
1 parent 3a502c4 commit 65b2c45

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exnn/base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def get_active_subnets(self, beta_threshold=0):
156156
if self.bn_flag:
157157
beta = self.output_layer.output_weights.numpy()
158158
else:
159-
subnet_norm = [self.subnet_blocks.subnets[i].moving_norm.numpy()[0] for i in range(self.numerical_input_num)]
159+
subnet_norm = [self.subnet_blocks.subnets[i].moving_norm.numpy()[0] for i in range(self.subnet_num)]
160160
categ_norm = [self.categ_blocks.categnets[i].moving_norm.numpy()[0]for i in range(self.categ_variable_num)]
161161
beta = self.output_layer.output_weights.numpy() * np.hstack([subnet_norm, categ_norm]).reshape([-1, 1])
162162
beta = beta * self.output_layer.output_switcher.numpy()

0 commit comments

Comments
 (0)