model_encoder

class sparx.model_encoder.EncodedModel(num_layers, layer_shapes, weights, biases, activation_functions)

Bases: object

Encoded model class.

class sparx.model_encoder.Framework(value)

Bases: Enum

Framework enum

KERAS = 1

PYTORCH = 2

class sparx.model_encoder.Model

Bases: object

Model class

static activation_to_str(activation) → str

Translate activation function to string

static get_keras_model_info(model: Model) → EncodedModel

Encode model information.

static transform(model: any, framework: Framework)

Transform model to encoded form.