Python BaseStep._did_process Examples

Example #1

File: loop.py Project: guillaume-chevalier/Neuraxle

    def _did_process(self, data_container: DataContainer,
                     context: ExecutionContext) -> DataContainer:
        """
        Reaugment the flattened data container.

        :param data_container: data container to then_unflatten
        :param context: execution context
        :return: data container
        """
        data_container = BaseStep._did_process(self, data_container, context)

        if self.then_unflatten:
            data_container.set_data_inputs(
                self._reaugment_list(data_container.data_inputs, self.len_di))
            data_container.set_expected_outputs(
                self._reaugment_list(data_container.expected_outputs,
                                     self.len_eo))
            self.len_di = []
            self.len_eo = []

        return data_container

Example #2

File: flow.py Project: rodcanada/Neuraxle

 def _did_process(self, data_container, context):
     data_container = BaseStep._did_process(self, data_container, context)
     return data_container.reduce_dim()