so_magic.som.factory

Complexity

Total Complexity

2

Size/Duplication

Total Lines	25
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	SelfOrganizingMapFactory.create()	0	10	2

import logging
import attr
from so_magic.utils import Subject
from .self_organising_map import SomTrainer, SelfOrganizingMap, NoFeatureVectorsError


logger = logging.getLogger(__name__)


@attr.s
class SelfOrganizingMapFactory:
    trainer = attr.ib(init=True, default=attr.Factory(SomTrainer.from_callable))
    subject = attr.ib(init=True, default=attr.Factory(lambda: Subject([])))

    def create(self, dataset, nb_cols, nb_rows, **kwargs):
        try:
            # run a backend algorithm and get a self-organising map representation object
            somoclu_map = self.trainer.infer_map(nb_cols, nb_rows, dataset, **kwargs)
            self.subject.state = somoclu_map
            self.subject.notify()
            return SelfOrganizingMap(somoclu_map, dataset.name)
        except NoFeatureVectorsError as exception:
            logger.info("%s Fire up an 'encode' command.", str(exception))
            raise exception