update check network

Author: antoinedemathelin

adapt/feature_based/_msda.py

@@ -287,4 +287,22 @@ def predict(self, X):
             Prediction of ``estimator_``.
         """
         X = check_one_array(X)
-        return self.estimator_.predict(self.encoder_.predict(X))
+        return self.estimator_.predict(self.predict_features(X))
+    
+    
+    def predict_features(self, X):
+        """
+        Return the encoded features of X.
+        
+        Parameters
+        ----------
+        X: array
+            input data
+            
+        Returns
+        -------
+        X_enc: array
+            predictions of encoder network
+        """
+        X = check_one_array(X)
+        return self.encoder_.predict(X)

adapt/utils.py

@@ -202,7 +202,8 @@ def check_network(network, copy=True,
     if copy:
         try:
             if hasattr(network, "input_shape"):
-                new_network = clone_model(network, input_tensors=network.input)
+                shape = network.input_shape[1:]
+                new_network = clone_model(network, input_tensors=Input(shape))
                 new_network.set_weights(network.get_weights())
             else:
                 new_network = clone_model(network)