[英]Rookie for Keras: how to load a pretrained MalConv model to predict in my data?
我想使用預訓練的MalConv.h5在自己的數據集中進行預測。
我在這個VM之后安裝了環境(Keras,tensorflow)
from numpy import loadtxt
from keras.models import load_model
import h5py
model = load_model('malconv.h5')
model.summary()
prediction =
model.predict('data/train/0014D73E9987A3FC3DA1055D912286B95929DC6D')
Print(prediction)
結果:
2019-10-30 08:06:09.605829: I tensorflow/core/platform/cpu_feature_guard.cc:137] 您的 CPU 支持此 TensorFlow 二進制文件未編譯使用的指令:SSE4.1 SSE4.2 AVX AVX2 FMA
如何獲得對數據文件的預測(預期結果):惡意軟件 1,良性 0?
您試圖預測的數據是什么? 看起來您正在嘗試訪問磁盤上的某些內容而不將其讀入 memory。
您需要將數據讀入一個變量,然后將其傳遞給 predict() 方法
我發現這篇文章可能有用,問題是您傳遞的是字符串,而不是文件
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.