我想使用预先训练的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?
您要预测的数据是什么?似乎您正在尝试访问磁盘上的某些内容而不将其读入内存。
您需要将数据读入变量,然后将其传递给predict()方法