dice_ml.model_interfaces package¶
Submodules¶
dice_ml.model_interfaces.base_model module¶
Module containing a template class as an interface to ML model. Subclasses implement model interfaces for different ML frameworks such as TensorFlow, PyTorch OR Sklearn. All model interface methods are in dice_ml.model_interfaces
-
class
dice_ml.model_interfaces.base_model.
BaseModel
(model=None, model_path='', backend='', func=None, kw_args=None)[source]¶ Bases:
object
dice_ml.model_interfaces.keras_tensorflow_model module¶
Module containing an interface to trained Keras Tensorflow model.
-
class
dice_ml.model_interfaces.keras_tensorflow_model.
KerasTensorFlowModel
(model=None, model_path='', backend='TF1', func=None, kw_args=None)[source]¶
dice_ml.model_interfaces.pytorch_model module¶
Module containing an interface to trained PyTorch model.
-
class
dice_ml.model_interfaces.pytorch_model.
PyTorchModel
(model=None, model_path='', backend='PYT', func=None, kw_args=None)[source]¶