【发布时间】:2020-02-25 22:22:50
【问题描述】:
我想使用预训练的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?
【问题讨论】:
标签: python tensorflow keras pycharm