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)
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)
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)
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):
Common base class for all non-exit exceptions.
Inherited Members
- builtins.Exception
- Exception
- builtins.BaseException
- with_traceback
- args
class
ModelConversionFailure(builtins.Exception):
Common base class for all non-exit exceptions.
Inherited Members
- builtins.Exception
- Exception
- builtins.BaseException
- with_traceback
- args
class
ModelConversionUnsupportedType(builtins.Exception):
Common base class for all non-exit exceptions.
Inherited Members
- builtins.Exception
- Exception
- builtins.BaseException
- with_traceback
- args
class
ModelConversionSourceFileNotPresent(builtins.Exception):
Common base class for all non-exit exceptions.
Inherited Members
- builtins.Exception
- Exception
- builtins.BaseException
- with_traceback
- args