簡體   English   中英

Keras 的新秀:如何加載預訓練的 MalConv model 以在我的數據中進行預測?

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM