Keras新秀:如何加载经过预训练的MalConv模型以在我的数据中进行预测?

问题描述 投票:0回答:1

我想使用预先训练的MalConv.h5在自己的数据集中进行预测。

我在此VM之后安装了环境(Keras,张量流)

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:我tensorflow / core / platform / cpu_feature_guard.cc:137]您的CPU支持该TensorFlow二进制文件未编译为使用的指令:SSE4.1 SSE4.2 AVX AVX2 FMA

如何获得对数据文件的预测(预期结果):恶意软件1,良性0?

python tensorflow keras pycharm
1个回答
0
投票

您要预测的数据是什么?似乎您正在尝试访问磁盘上的某些内容而不将其读入内存。

您需要将数据读入变量,然后将其传递给predict()方法

© www.soinside.com 2019 - 2024. All rights reserved.