Struct owlyshield_ransom::predictions::prediction_malware::TfLiteMalware

pub struct TfLiteMalware {
    model: Model,
    means: Vec<f32>,
    stdvs: Vec<f32>,
}

A record to describe a tflite model

Fields

model: Model

means: Vec<f32>

Needed by Standard Scaling and set to MEANS

stdvs: Vec<f32>

Needed by Standard Scaling and set to STDVS

Implementations

impl TfLiteMalware

pub fn new() -> TfLiteMalware

pub fn make_prediction(&self, predmtrx: &VecvecCapped<f32>) -> f32

Make a prediction on the sequence predmtrx. The prediction can be costly. The model input tensor dimensions are (None, PREDMTRXCOLS) and is dimensioned accordingly by the InterpreterBuilder. The model returns only the last prediction (it does not returns sequences).

fn standardize(&self, predmtrx: &VecvecCapped<f32>) -> VecvecCapped<f32>

Standard Scaling of the input vectors with MEANS and STDVS.