@@ -469,6 +469,8 @@ def fit(self, X, y, Xt=None, yt=None, domains=None, **fit_params):
         else:
             Xt = check_array(Xt, ensure_2d=True, allow_nd=True)
         set_random_seed(self.random_state)
+        
+        self.n_features_in_ = X.shape[1]
 
         if hasattr(self, "fit_weights"):
             if self.verbose:
@@ -1663,4 +1665,4 @@ def _initialize_networks(self):
                                                 name="discriminator")
 
     def _initialize_pretain_networks(self):
-        pass
+        pass