class_docstr: Implementation of container for custom model, which is presented as function with
  input train_data(np.array), test_data(np.array), parameters(dict)
  output type specification DataTypesEnum (string - 'ts', 'table', 'image', 'text')
  into parameters dictionary {'model_predict': function, 'model_fit': function}
class_docstr_tok: ["Implementation","of","container","for","custom","model",",","which","is","presented","as","function","with","input","train_data","(","np.array",")",",","test_data","(","np.array",")",",","parameters","(","dict",")","output","type","specification","DataTypesEnum","(","string","-","'ts","'",",","'table","'",",","'image","'",",","'text","'",")","into","parameters","dictionary","{","'model_predict","'",":","function",",","'model_fit","'",":","function","}"]
code: def predict(self, input_data):
      try:
          predict, output_type = self.model_predict(
              self.fitted_model,
              input_data.idx,
              input_data.features,
              self.params,
          )
          if (
              input_data.data_type == DataTypesEnum.ts
              and input_data.target is not None
              and len(input_data.target.shape) > 1
          ):
              # change target after custom model is processed
              input_data.target = input_data.target[:, 0]
          output_type = DataTypesEnum[output_type]
      except Exception as e:
          raise TypeError(
              f"{e}\nInput model has incorrect behaviour. Check type hints for model: \
                                      Callable[[any, np.array, dict], np.array, str]"
          )
      output_data = self._convert_to_output(
          input_data, predict=predict, data_type=output_type
      )
      return output_data
  
file_path: FEDOT/fedot/core/operations/evaluation/operation_implementations/models/custom_model.py
imports: from typing import Callable, Optional
  from fedot.core.operations.evaluation.operation_implementations.implementation_interfaces import ModelImplementation
  from fedot.core.operations.operation_parameters import OperationParameters
  from fedot.core.repository.dataset_types import DataTypesEnum
label_desc: Use image node_id 5 for calling the CustomModelImplementation obj's underlying member method code with example usage: obj.predict(input_data) and returns: output_data