Module `skplumber.consts`

Expand source code

from enum import Enum


class ProblemType(Enum):
    CLASSIFICATION = "classification"
    REGRESSION = "regression"

    @classmethod
    def has_value(cls, value):
        return value in set(item.value for item in cls)


class OptimizationDirection(Enum):
    MINIMIZE = 1
    MAXIMIZE = 2


class PrimitiveType(Enum):
    PREPROCESSOR = 1
    TRANSFORMER = 2
    REGRESSOR = 3
    CLASSIFIER = 4

Classes

class OptimizationDirection (value, names=None, *, module=None, qualname=None, type=None, start=1)

An enumeration.

Expand source code

class OptimizationDirection(Enum):
    MINIMIZE = 1
    MAXIMIZE = 2

Ancestors

enum.Enum

Class variables

var MAXIMIZE
var MINIMIZE

class PrimitiveType (value, names=None, *, module=None, qualname=None, type=None, start=1)

An enumeration.

Expand source code

class PrimitiveType(Enum):
    PREPROCESSOR = 1
    TRANSFORMER = 2
    REGRESSOR = 3
    CLASSIFIER = 4

Ancestors

enum.Enum

Class variables

var CLASSIFIER
var PREPROCESSOR
var REGRESSOR
var TRANSFORMER

class ProblemType (value, names=None, *, module=None, qualname=None, type=None, start=1)

An enumeration.

Expand source code

class ProblemType(Enum):
    CLASSIFICATION = "classification"
    REGRESSION = "regression"

    @classmethod
    def has_value(cls, value):
        return value in set(item.value for item in cls)

Ancestors

enum.Enum

Class variables

var CLASSIFICATION
var REGRESSION