Update stacked_tabnet.py (#19)

Fixing #17
titu1994 · Apr 22, 2021 · f35d964 · f35d964
1 parent 9a76671
commit f35d964
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tabnet/stacked_tabnet.py b/tabnet/stacked_tabnet.py
@@ -348,6 +348,6 @@ def __init__(self, feature_columns,
         self.regressor = tf.keras.layers.Dense(num_regressors, use_bias=False)
 
     def call(self, inputs, training=None):
-        self.activations = self.tabnet(inputs, training=training)
+        self.activations = self.stacked_tabnet(inputs, training=training)
         out = self.regressor(self.activations)
         return out