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

时间:2019-10-30 15:09:57

标签: python tensorflow keras pycharm

我想使用经过训练的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?

1 个答案:

答案 0 :(得分:0)

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

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

我发现this文章可能有用,事实是您传递的是字符串,而不是文件