Source code for explauto.interest_model.interest_model

from abc import ABCMeta, abstractmethod

from . import interest_models

[docs]class InterestModel(object):
    __metaclass__ = ABCMeta

    def __init__(self, expl_dims):
        self.expl_dims = expl_dims

    @classmethod
[docs]    def from_configuration(cls, conf, expl_dims, im_name, config_name='default'):
        im_cls, im_configs = interest_models[im_name]
        return im_cls(conf, expl_dims, **im_configs[config_name])

    @abstractmethod
[docs]    def sample(self):
        pass

    @abstractmethod
[docs]    def update(self, xy, ms):
        pass