我想使用经过训练的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:我 tensorflow / core / platform / cpu_feature_guard.cc:137]您的CPU支持 该TensorFlow二进制文件未编译为使用的指令: SSE4.1 SSE4.2 AVX AVX2 FMA
如何获取对数据文件的预测(预期结果):恶意软件1,良性0?
答案 0 :(得分:0)
您要预测的数据是什么? 看来您正在尝试访问磁盘上的某些内容而不将其读入内存。
您需要将数据读取到变量中,然后将其传递给predict()方法
我发现this文章可能有用,事实是您传递的是字符串,而不是文件