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 or PyTorch. All model interface methods are in dice_ml.model_interfaces

class dice_ml.model_interfaces.base_model.BaseModel(model=None, model_path='', backend='')[source]

Bases: object

get_gradient()[source]
get_output()[source]
load_model()[source]

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')[source]

Bases: dice_ml.model_interfaces.base_model.BaseModel

get_gradient(input)[source]
get_output(input_tensor, training=False)[source]
load_model()[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')[source]

Bases: dice_ml.model_interfaces.base_model.BaseModel

get_gradient(input)[source]
get_output(input_tensor)[source]
load_model()[source]
set_eval_mode()[source]

Module contents