Documentation for Defaultmodelhandler Module

DefaultModelHandler

Bases: ModelHandler

Source code in nebula/core/situationalawareness/discovery/modelhandlers/defaultmodelhandler.py

class DefaultModelHandler(ModelHandler):
    def __init__(self):
        self.model = None
        self.rounds = 0
        self.round = 0
        self.epochs = 0
        self.model_lock = Locker(name="model_lock")
        self.params_lock = Locker(name="param_lock")
        self._nm: FederationConnector = None

    def set_config(self, config):
        """
        Args:
            config[0] -> total rounds
            config[1] -> current round
            config[2] -> epochs
            config[3] -> FederationConnector
        """
        self.params_lock.acquire()
        self.rounds = config[0]
        if config[1] > self.round:
            self.round = config[1]
        self.epochs = config[2]
        if not self._nm:
            self._nm = config[3]
        self.params_lock.release()

    def accept_model(self, model):
        return True

    async def get_model(self, model):
        """
        Returns:
            model with default weights
        """
        (sm, _, _) = await self._nm.engine.cm.propagator.get_model_information(None, "initialization", init=True)
        return (sm, self.rounds, self.round, self.epochs)

    def pre_process_model(self):
        """
        no pre-processing defined
        """
        pass

get_model(model) async

Returns:

Type	Description
	model with default weights

Source code in nebula/core/situationalawareness/discovery/modelhandlers/defaultmodelhandler.py

async def get_model(self, model):
    """
    Returns:
        model with default weights
    """
    (sm, _, _) = await self._nm.engine.cm.propagator.get_model_information(None, "initialization", init=True)
    return (sm, self.rounds, self.round, self.epochs)

pre_process_model()

no pre-processing defined

Source code in nebula/core/situationalawareness/discovery/modelhandlers/defaultmodelhandler.py

def pre_process_model(self):
    """
    no pre-processing defined
    """
    pass

set_config(config)

Source code in nebula/core/situationalawareness/discovery/modelhandlers/defaultmodelhandler.py

def set_config(self, config):
    """
    Args:
        config[0] -> total rounds
        config[1] -> current round
        config[2] -> epochs
        config[3] -> FederationConnector
    """
    self.params_lock.acquire()
    self.rounds = config[0]
    if config[1] > self.round:
        self.round = config[1]
    self.epochs = config[2]
    if not self._nm:
        self._nm = config[3]
    self.params_lock.release()