so_magic.data.backend.panda_handling.df_discretization

Complexity

Total Complexity

8

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	PDBinnerFactory.equal_length_binner()	0	2	1
A	DFSameLengthBinning.bin()	0	2	1
A	PDBinnerFactory.create_binner()	0	2	1
A	BasePDBinner.register_as_subclass()	0	6	1
A	DFQBinning.bin()	0	2	1
A	BasePDBinner.create()	0	5	2
A	PDBinnerFactory.quantisized_binner()	0	2	1

from abc import ABC
import pandas as pd

from data.discretization import BaseBinner, BinnerFactory


class BasePDBinner(BaseBinner, ABC):

    @classmethod
    def register_as_subclass(cls, binner_type):
        def wrapper(subclass):
            cls.subclasses[binner_type] = subclass
            return subclass
        return wrapper

    @classmethod
    def create(cls, binner_type, *args, **kwargs):
        if binner_type not in cls.subclasses:
            raise ValueError('Bad "BinnerFactory Backend type" type \'{}\''.format(binner_type))
        return cls.subclasses[binner_type](*args, **kwargs)


@BaseDFBinner.register_as_subclass('same-length')

The variable BaseDFBinner does not seem to be defined.

class DFSameLengthBinning(BasePDBinner):

    def bin(self, values, nb_bins):
        return pd.cut(values, nb_bins)

@BaseDFBinner.register_as_subclass('quantisized')

The variable BaseDFBinner does not seem to be defined.

class DFQBinning(BasePDBinner):

    def bin(self, values, nb_bins):
        return pd.qcut(values, nb_bins)


@BinnerFactory.register_as_subclass('pandas')
class PDBinnerFactory(BinnerFactory):

    def equal_length_binner(self, *args, **kwargs) -> DFSameLengthBinning:
        return DFSameLengthBinning()

    def quantisized_binner(self, *args, **kwargs) -> DFQBinning:
        return DFQBinning()

    def create_binner(self, *args, **kwargs) -> BasePDBinner:
        BasePDBinner.create(*args, **kwargs)