Esempio n. 1
0
 def apply_function(self):
     '''Cf. `PiStage` docs.'''
     # TODO: do we need following line? Isn't this handled universally
     # by the base class (in PiStage's apply)?
     self.data.data_specs = self.output_specs
     # reset weights to initial weights prior to downstream stages running
     for container in self.data:
         vectorizer.set(container['initial_weights'],
                        out=container['weights'])
Esempio n. 2
0
    def setup_function(self):
        """Setup the stage"""

        # set the correct data mode
        self.data.data_specs = self.calc_specs
        for container in self.data:
            container['calculated_pid'] = np.empty((container.size),
                                                   dtype=FTYPE)
            container['original_pid'] = np.empty((container.size), dtype=FTYPE)
            vectorizer.set(container['pid'], out=container['original_pid'])
Esempio n. 3
0
 def apply_function(self):
     for container in self.data:
         vectorizer.set(container['initial_weights'],
                        out=container['weights'])
Esempio n. 4
0
 def apply_function(self):
     for container in self.data:
         # set the pid value to the calculated one
         vectorizer.set(container['calculated_pid'], out=container['pid'])