def run(self, obj, config): mmb = MaliciousMacroBot() mmb.mmb_init_model() mmb.set_model_paths(benign_path=None, malicious_path=None, model_path=self.model) fc =(obj.filedata.read()) result = mmb.mmb_predict(fc, datatype='filecontents') json = mmb.mmb_prediction_to_json(result)[0] for k,v in json.iteritems(): if k == 'prediction': self._add_result("Prediction", v, {"name": k}) for k,v in json.iteritems(): if k != 'prediction': self._add_result("Features", v, {"name": k})
def run(self, obj, config): mmb = MaliciousMacroBot() mmb.mmb_init_model() mmb.set_model_paths(benign_path=None, malicious_path=None, model_path=self.model) f = tempfile.NamedTemporaryFile() f.write(obj.filedata.read()) result = mmb.mmb_predict(f.name, datatype='filepath') f.close() json = mmb.mmb_prediction_to_json(result)[0] for k, v in json.iteritems(): self._add_result("Prediction", k, {"value": v})