wafamole.models package

Submodules

wafamole.models.keras_model module

wafamole.models.model module

Abstract machine learning model.

class wafamole.models.model.Model

Bases: object

Abstract machine learning model wrapper.

classify(value: object)

It returns the probability of belonging to a particular class. It calls the extract_features function on the input value to produce a feature vector.

Parameters:value (object) – Input value Returns:the confidence of the malicious class. Return type:float

extract_features(value: object)

It extract a feature vector from the input object.

Parameters:value (object) – An input point that belongs to the input space of the wrapped model. Returns:array containing the feature vector of the input value. Return type:feature_vector (numpy ndarray) Raises:NotImplementedError – this method needs to be implemented

wafamole.models.sklearn_model module

Module contents