tuenv2kProPTIT / btlMalwareDetection

MALWARE DETECTION

PREDICT

import numpy as np
from src import pipeline_stat
model = pipeline_stat.Model('./checkpoint/stat/', version=3)

ans = model.predict_from_path('./malware.exe')

print(ans)
#{'y_pred': 0, 'values': 'MALWARE', 'probs': array([[0.84063112, 0.15936888]])}