wallaroo.ModelConversion

 1from enum import Enum
 2from typing import Any, Dict, List, NamedTuple, Union
 3
 4
 5class ModelConversionInputType(Enum):
 6    Float16 = "float16"
 7    Float32 = "float32"
 8    Float64 = "float64"
 9    Int16 = "int16"
10    Int32 = "int32"
11    Int64 = "int64"
12    UInt8 = "uint8"
13    UInt16 = "uint16"
14    UInt32 = "uint32"
15    UInt64 = "uint64"
16    Boolean = "bool"
17    Double = "double"
18
19
20class ConvertKerasArguments(NamedTuple):
21    name: str
22    comment: Union[str, None]
23    input_type: ModelConversionInputType
24    dimensions: List[Union[None, int, float]]
25
26    def to_dict(self) -> Dict[str, Any]:
27        base = self._asdict()
28        base["input_type"] = self.input_type.value
29        return base
30
31
32class ConvertSKLearnArguments(NamedTuple):
33    name: str
34    number_of_columns: int
35    input_type: ModelConversionInputType
36    comment: Union[str, None]
37
38    def to_dict(self) -> Dict[str, Any]:
39        base = self._asdict()
40        base["input_type"] = self.input_type.value
41        return base
42
43
44class ConvertXGBoostArgs(NamedTuple):
45    name: str
46    number_of_columns: int
47    input_type: ModelConversionInputType
48    comment: Union[str, None]
49
50    def to_dict(self) -> Dict[str, Any]:
51        base = self._asdict()
52        base["input_type"] = self.input_type.value
53        return base
54
55
56ModelConversionArguments = Union[
57    ConvertKerasArguments, ConvertSKLearnArguments, ConvertXGBoostArgs
58]
59
60
61class ModelConversionSource(Enum):
62    KERAS = "keras"
63    XGBOOST = "xgboost"
64    SKLEARN = "sklearn"
65
66
67class ModelConversionGenericException(Exception):
68    pass
69
70
71class ModelConversionFailure(Exception):
72    pass
73
74
75class ModelConversionUnsupportedType(Exception):
76    pass
77
78
79class ModelConversionSourceFileNotPresent(Exception):
80    pass
class ModelConversionInputType(enum.Enum):
 6class ModelConversionInputType(Enum):
 7    Float16 = "float16"
 8    Float32 = "float32"
 9    Float64 = "float64"
10    Int16 = "int16"
11    Int32 = "int32"
12    Int64 = "int64"
13    UInt8 = "uint8"
14    UInt16 = "uint16"
15    UInt32 = "uint32"
16    UInt64 = "uint64"
17    Boolean = "bool"
18    Double = "double"

An enumeration.

Float16 = <ModelConversionInputType.Float16: 'float16'>
Float32 = <ModelConversionInputType.Float32: 'float32'>
Float64 = <ModelConversionInputType.Float64: 'float64'>
Int16 = <ModelConversionInputType.Int16: 'int16'>
Int32 = <ModelConversionInputType.Int32: 'int32'>
Int64 = <ModelConversionInputType.Int64: 'int64'>
UInt8 = <ModelConversionInputType.UInt8: 'uint8'>
UInt16 = <ModelConversionInputType.UInt16: 'uint16'>
UInt32 = <ModelConversionInputType.UInt32: 'uint32'>
UInt64 = <ModelConversionInputType.UInt64: 'uint64'>
Boolean = <ModelConversionInputType.Boolean: 'bool'>
Double = <ModelConversionInputType.Double: 'double'>
Inherited Members
enum.Enum
name
value
class ConvertKerasArguments(typing.NamedTuple):
21class ConvertKerasArguments(NamedTuple):
22    name: str
23    comment: Union[str, None]
24    input_type: ModelConversionInputType
25    dimensions: List[Union[None, int, float]]
26
27    def to_dict(self) -> Dict[str, Any]:
28        base = self._asdict()
29        base["input_type"] = self.input_type.value
30        return base

ConvertKerasArguments(name, comment, input_type, dimensions)

ConvertKerasArguments( name: str, comment: Optional[str], input_type: wallaroo.ModelConversion.ModelConversionInputType, dimensions: List[Union[NoneType, int, float]])

Create new instance of ConvertKerasArguments(name, comment, input_type, dimensions)

name: str

Alias for field number 0

comment: Optional[str]

Alias for field number 1

Alias for field number 2

dimensions: List[Union[NoneType, int, float]]

Alias for field number 3

def to_dict(self) -> Dict[str, Any]:
27    def to_dict(self) -> Dict[str, Any]:
28        base = self._asdict()
29        base["input_type"] = self.input_type.value
30        return base
Inherited Members
builtins.tuple
index
count
class ConvertSKLearnArguments(typing.NamedTuple):
33class ConvertSKLearnArguments(NamedTuple):
34    name: str
35    number_of_columns: int
36    input_type: ModelConversionInputType
37    comment: Union[str, None]
38
39    def to_dict(self) -> Dict[str, Any]:
40        base = self._asdict()
41        base["input_type"] = self.input_type.value
42        return base

ConvertSKLearnArguments(name, number_of_columns, input_type, comment)

ConvertSKLearnArguments( name: str, number_of_columns: int, input_type: wallaroo.ModelConversion.ModelConversionInputType, comment: Optional[str])

Create new instance of ConvertSKLearnArguments(name, number_of_columns, input_type, comment)

name: str

Alias for field number 0

number_of_columns: int

Alias for field number 1

Alias for field number 2

comment: Optional[str]

Alias for field number 3

def to_dict(self) -> Dict[str, Any]:
39    def to_dict(self) -> Dict[str, Any]:
40        base = self._asdict()
41        base["input_type"] = self.input_type.value
42        return base
Inherited Members
builtins.tuple
index
count
class ConvertXGBoostArgs(typing.NamedTuple):
45class ConvertXGBoostArgs(NamedTuple):
46    name: str
47    number_of_columns: int
48    input_type: ModelConversionInputType
49    comment: Union[str, None]
50
51    def to_dict(self) -> Dict[str, Any]:
52        base = self._asdict()
53        base["input_type"] = self.input_type.value
54        return base

ConvertXGBoostArgs(name, number_of_columns, input_type, comment)

ConvertXGBoostArgs( name: str, number_of_columns: int, input_type: wallaroo.ModelConversion.ModelConversionInputType, comment: Optional[str])

Create new instance of ConvertXGBoostArgs(name, number_of_columns, input_type, comment)

name: str

Alias for field number 0

number_of_columns: int

Alias for field number 1

Alias for field number 2

comment: Optional[str]

Alias for field number 3

def to_dict(self) -> Dict[str, Any]:
51    def to_dict(self) -> Dict[str, Any]:
52        base = self._asdict()
53        base["input_type"] = self.input_type.value
54        return base
Inherited Members
builtins.tuple
index
count
class ModelConversionSource(enum.Enum):
62class ModelConversionSource(Enum):
63    KERAS = "keras"
64    XGBOOST = "xgboost"
65    SKLEARN = "sklearn"

An enumeration.

KERAS = <ModelConversionSource.KERAS: 'keras'>
XGBOOST = <ModelConversionSource.XGBOOST: 'xgboost'>
SKLEARN = <ModelConversionSource.SKLEARN: 'sklearn'>
Inherited Members
enum.Enum
name
value
class ModelConversionGenericException(builtins.Exception):
68class ModelConversionGenericException(Exception):
69    pass

Common base class for all non-exit exceptions.

Inherited Members
builtins.Exception
Exception
builtins.BaseException
with_traceback
args
class ModelConversionFailure(builtins.Exception):
72class ModelConversionFailure(Exception):
73    pass

Common base class for all non-exit exceptions.

Inherited Members
builtins.Exception
Exception
builtins.BaseException
with_traceback
args
class ModelConversionUnsupportedType(builtins.Exception):
76class ModelConversionUnsupportedType(Exception):
77    pass

Common base class for all non-exit exceptions.

Inherited Members
builtins.Exception
Exception
builtins.BaseException
with_traceback
args
class ModelConversionSourceFileNotPresent(builtins.Exception):
80class ModelConversionSourceFileNotPresent(Exception):
81    pass

Common base class for all non-exit exceptions.

Inherited Members
builtins.Exception
Exception
builtins.BaseException
with_traceback
args