tracking_policy_agendas.classifiers.meta_sklearn_clf

Complexity

Total Complexity

11

Size/Duplication

Total Lines	50
Duplicated Lines	66 %

Test Coverage

Coverage

68.18%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	MetaSkLearnClf.load_model()	13	13	4
A	MetaSkLearnClf.__init__()	3	3	3
A	MetaSkLearnClf.save_model()	14	14	4

"""
Abstract class for SKLearn Classifiers

....................................................................................................
MIT License
Copyright (c) 2021-2023 AUT Iran, Mohammad H Forouhesh
Copyright (c) 2021-2022 MetoData.ai, Mohammad H Forouhesh
....................................................................................................
This module abstracts sklearn classifier.
"""

import os
import pickle
from .meta_clf import MetaClf


class MetaSkLearnClf(MetaClf):

This code seems to be duplicated in your project.

Missing class docstring

    def __init__(self, classifier_instance, **kwargs):
        super().__init__(classifier_instance=classifier_instance, **kwargs)
        if 'load_path' in kwargs and kwargs['load_path'] is not None: self.load_model(kwargs['load_path'])

More than one statement on a single line

This line is too long as per the coding-style (106/100).

    def load_model(self, load_path: str):
        """
        A tool to load model from disk.
        :param load_path:   Model path.
        :return:            None
        """

        loading_prep = lambda string: f'model_dir/{load_path}/{string}'
        self.emb.load(loading_prep('emb.pkl'))
        with open(loading_prep('model.pkl'), 'rb') as f:

Variable name "f" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

            self.clf = pickle.load(f)
        with open(loading_prep('scaler.pkl'), 'rb') as f:

Variable name "f" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

            self.scaler = pickle.load(f)

    def save_model(self, save_path: str):
        """
        A tool to save model to disk
        :param save_path:   Saving path.
        :return:            None.
        """

        os.makedirs(f'model_dir/{save_path}', exist_ok=True)
        saving_prep = lambda string: f'model_dir/{save_path}/{string}'
        self.emb.save(saving_prep('emb.pkl'))
        with open(saving_prep('scaler.pkl'), 'wb') as f:

Variable name "f" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

            pickle.dump(self.scaler, f, pickle.HIGHEST_PROTOCOL)
        with open(saving_prep('model.pkl'), 'wb') as f:

Variable name "f" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

            pickle.dump(self.clf, f, pickle.HIGHEST_PROTOCOL)