【问题标题】:Rookie for Keras: how to load a pretrained MalConv model to predict in my data?Keras 新手:如何加载预训练的 MalConv 模型以在我的数据中进行预测?
【发布时间】: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


    【解决方案1】:

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

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

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

    【讨论】:

    • 感谢您的回复!数据在我的磁盘上,数据类型是 .exe(二进制)文件。我认为它与变量等一般数据不同,因为该模型用于 exe 文件的恶意软件检测。
    • 查看我链接的文章,听起来像你的用例
    猜你喜欢
    • 2017-07-16
    • 1970-01-01
    • 2017-03-13
    • 2018-04-15
    • 2020-05-04
    • 1970-01-01
    • 2020-02-27
    • 2020-05-12
    • 1970-01-01
    相关资源
    最近更新 更多