Source code for openmdao.lib.components.metamodel

""" Metamodel provides basic Meta Modeling capability"""

# pylint: disable-msg=E0611,F0401
from numpy import array
from openmdao.lib.datatypes.api import Instance, ListStr, Event, \
     List, Str, Dict
from enthought.traits.trait_base import not_none
from enthought.traits.has_traits import _clone_trait

from openmdao.main.api import Component, Case
from openmdao.main.interfaces import IComponent, ISurrogate, ICaseRecorder, \
from openmdao.main.uncertain_distributions import UncertainDistribution, \
from openmdao.main.mp_support import has_interface

[docs]class MetaModel(Component): """ A component that provides general Meta Modeling capability. See the Standard Library Reference for additional information on the :ref:`MetaModel` component.""" # pylint: disable-msg=E1101 model = Instance(Component, allow_none=True, desc='Socket for the Component or Assembly being ' 'encapsulated.') includes = ListStr(iotype='in', desc='A list of names of variables to be included ' 'in the public interface.') excludes = ListStr(iotype='in', desc='A list of names of variables to be excluded ' 'from the public interface.') warm_start_data = Instance(ICaseIterator,iotype="in", desc="CaseIterator containing cases to use as " "initial training data. When this is set, all " "previous training data is cleared, and replaced " "with data from this CaseIterator") surrogate = Dict(key_train=Str, value_trait=ISurrogate, allow_none=True, desc='An dictionary provides a mapping between variables and ' 'surrogate models for each output. The "default" ' 'key must be given. It is the default surrogate model for all ' 'outputs. Any specific surrogate models can be ' 'specifed by a key with the desired variable name.' ) recorder = Instance(ICaseRecorder, desc = 'Records training cases') # when fired, the next execution will train the metamodel train_next = Event() #when fired, the next execution will reset all training data reset_training_data = Event() def __init__(self, *args, **kwargs): super(MetaModel, self).__init__(*args, **kwargs) self._current_model_traitnames = set() self._surrogate_info = {} self._surrogate_input_names = [] self._training_input_history = [] self._train = False self._new_train_data = False # the following line will work for classes that inherit from MetaModel # as long as they declare their traits in the class body and not in # the __init__ function. If they need to create traits dynamically # during initialization they'll have to provide the value of # _mm_class_traitnames self._mm_class_traitnames = set(self.traits(iotype=not_none).keys()) def _train_next_fired(self): self._train = True self._new_train_data = True def _reset_training_data_fired(self): self._training_input_history = [] self.update_model(self.model, self.model) def _warm_start_data_changed(self,oldval,newval): self.reset_training_date = True #build list of inputs inputs = [] for case in newval.get_iter(): self.recorder.record(case) inputs = [] for inp_name in self._surrogate_input_names: inp_val = None #TODO: Fix case object, so it has a get_input method to clean up this loop var_name = "%s.%s"%(,inp_name) for name,val in case.items(iotype='in'): if name == var_name: inp_val = val break if inp_val is not None: inputs.append(inp_val) else: self.raise_exception('The variable "%s" was not ' 'found as an input in one of the cases provided ' 'for warm_start_data.'%var_name, ValueError) #print "inputs", inputs self._training_input_history.append(inputs) for output_name in self.list_outputs_from_model(): #grab value from case data output_val = None var_name = "%s.%s"%(,output_name) for name,val in case.items(iotype='out'): if name==var_name: output_val = val break if output_val is not None: # save to training output history #print output_name,":",output_val self._surrogate_info[output_name][1].append(output_val) else: self.raise_exception('The output "%s" was not found ' 'in one of the cases provided for ' 'warm_start_data'%var_name, ValueError) self._new_train_data = True
[docs] def execute(self): """If the training flag is set, train the metamodel. Otherwise, predict outputs. """ if self._train: if self.model: try: inputs = self.update_model_inputs() #print '%s training with inputs: %s' % (self.get_pathname(), inputs) except Exception as err: #self.raise_exception("training failed: %s" % str(err), type(err)) pass else: #if no exceptions are generated, save the data self._training_input_history.append(inputs) self.update_outputs_from_model() case_outputs = [] for name, tup in self._surrogate_info.items(): surrogate, output_history = tup case_outputs.append(('.'.join([,name]), output_history[-1])) # save the case, making sure to add out name to the local input name since # this Case is scoped to our parent Assembly case_inputs = [('.'.join([,name]),val) for name,val in zip(self._surrogate_input_names, inputs)] self.recorder.record(Case(inputs=case_inputs, outputs=case_outputs)) else: self.raise_exception("MetaModel object must have a model!", RuntimeError) self._train = False else: #print '%s predicting' % self.get_pathname() if self._new_train_data: for name,tup in self._surrogate_info.items(): surrogate, output_history = tup surrogate.train(self._training_input_history, output_history) self._new_train_data = False input_values = array([getattr(self, name) for name in self._surrogate_input_names]) for name, tup in self._surrogate_info.items(): surrogate = tup[0] predicted = surrogate.predict(input_values) # copy output to boudary setattr(self, name, predicted)
[docs] def invalidate_deps(self, compname=None, varnames=None, force=False): if compname: # we were called from our model, which expects to be in an Assembly return super(MetaModel, self).invalidate_deps(varnames=varnames)
[docs] def child_invalidated(self, childname, outs=None, force=False): pass
[docs] def exec_counts(self, compnames): # we force the run on our model, so it doesn't matter what we tell it the exec counts are return [0 for n in compnames]
def _model_changed(self, oldmodel, newmodel): self.update_model(oldmodel, newmodel)
[docs] def update_model(self, oldmodel, newmodel): """called whenever the model variable is set.""" # TODO: check for pre-connected traits on the new model # TODO: disconnect traits corresponding to old model (or leave them if the new model has the same ones?) # TODO: check for nested MMs? Is this a problem? # TODO: check for name collisions between MetaModel class traits and traits from model if newmodel is not None and not has_interface(newmodel, IComponent): self.raise_exception('model of type %s does not implement the IComponent interface' % type(newmodel).__name__, TypeError) if not self.surrogate: self.raise_exception("surrogate must be set before the model or any includes/excludes of variables", RuntimeError) new_model_traitnames = set() self._surrogate_input_names = [] self._taining_input_history = [] self._surrogate_info = {} # remove traits promoted from the old model for name in self._current_model_traitnames: if self.parent: self.parent.disconnect('.'.join([,name])) self.remove_trait(name) if newmodel: # query for inputs traitdict = newmodel._alltraits(iotype='in') for name,trait in traitdict.items(): if self._eligible(name): self._surrogate_input_names.append(name) if name not in self._mm_class_traitnames: self.add_trait(name, _clone_trait(trait)) new_model_traitnames.add(name) setattr(self, name, getattr(newmodel, name)) # now outputs traitdict = newmodel._alltraits(iotype='out') for name,trait in traitdict.items(): if self._eligible(name): try: surrogate = self.surrogate[name] except KeyError: try: surrogate = self.surrogate['default'] except KeyError: self.raise_exception("No default surrogate model was" " specified. Either specify a default, or specify a " "surrogate model for all outputs",ValueError) trait_type = surrogate.get_uncertain_value(1.0).__class__() self.add_trait(name, Instance(trait_type, iotype='out', desc=trait.desc)) self._surrogate_info[name] = (surrogate.__class__(), []) # (surrogate,output_history) new_model_traitnames.add(name) setattr(self, name, surrogate.get_uncertain_value(getattr(newmodel,name))) newmodel.parent = self = 'model' self._current_model_traitnames = new_model_traitnames
[docs] def update_inputs(self, compname, varnames): if compname != 'model': self.raise_exception("cannot update inputs for child named '%s'" % compname) self.model.set_valid(varnames, True)
[docs] def update_model_inputs(self): """Copy the values of the MetaModel's inputs into the inputs of the model. Returns the values of the inputs. """ input_values = [] for name in self._surrogate_input_names: inp = getattr(self, name) input_values.append(inp) setattr(self.model, name, inp) return input_values
[docs] def update_outputs_from_model(self): """Copy output values from the model into the MetaModel's outputs, and if training, save the output associated with surrogate. """ for name in self.list_outputs_from_model(): out = getattr(self.model, name) setattr(self, name, self._surrogate_info[name][0].get_uncertain_value(out)) if self._train: self._surrogate_info[name][1].append(out) # save to training output history
[docs] def list_inputs_to_model(self): """Return the list of names of public inputs that correspond to model inputs. """ return self._surrogate_input_names
[docs] def list_outputs_from_model(self): """Return the list of names of public outputs that correspond to model outputs. """ return list(set(self.list_outputs())-self._mm_class_traitnames)
def _includes_changed(self, old, new): if self.excludes and new is not None: self.__dict__['includes'] = old self.raise_exception("includes and excludes are mutually exclusive", RuntimeError) self.update_model(self.model, self.model) def _excludes_changed(self, old, new): if self.includes and new is not None: self.__dict__['excludes'] = old self.raise_exception("includes and excludes are mutually exclusive", RuntimeError) self.update_model(self.model, self.model) def _eligible(self, name): """Return True if the named trait is not excluded from the public interface based on the includes and excludes lists. """ if name in self._mm_class_traitnames: return False if self.includes and name not in self.includes: return False elif self.excludes and name in self.excludes: return False return True
OpenMDAO Home