Scikeras Tutorial - 4: Input transformer and MIMOEstimator

Input_Reshaper.py

def input_reshaper(X):
    return [X[:,:-10].reshape(X.shape[0],28,28,1), X[:,-10:]]

MIMOEstimator.py

class MIMOEstimator(BaseWrapper):

  @property
  def feature_encoder(self):
      return FunctionTransformer(
          func=input_reshaper,
      )

  @staticmethod
  def scorer(X, #should be y_true according to documentation and would be changed in next release
             y, #should be y_pred according to documentation and would be changed in next release
             **kwargs) -> float:
    y_pred_caps, y_pred_recons = y[:,:10],y[:,10:]
    y_caps, y_recons = X[:,:10],X[:,10:]

    return accuracy_score(y_caps, y_pred_caps)
  
  @property
  def target_encoder(self):
      return MultiOutputTransformer()