复现EMBER数据集问题解决

执行训练脚本时只能识别到一种样本

features.py里这一行缺一个方括号,加上就好

# 处理入口节区名称特征
        entry_name_hashed = FeatureHasher(50, input_type="string").transform([[raw_obj['entry']]]).toarray()[0]
        # entry_name_hashed = FeatureHasher(50, input_type="string").transform([raw_obj['entry']]).toarray()[0]

加上之后,发现问题解决不了,其实是因为在运行代码时用的是conda环境中的egg里的代码,要进入环境的目录,解压、改代码、重新压缩才行

# 创建临时目录
mkdir -p /tmp/ember_fix

# 解压egg文件到临时目录
cd /tmp/ember_fix
unzip /home/qy/miniconda3/envs/ember/lib/python3.8/site-packages/ember-0.1.0-py3.8.egg

# 编辑features.py文件
nano ember/features.py
# 找到并修改那一行代码

# 重新打包
zip -r /home/qy/miniconda3/envs/ember/lib/python3.8/site-packages/ember-0.1.0-py3.8.egg ember/*

# 返回项目目录
cd /mnt/e/malware/ember-master

# 重新运行训练脚本
python scripts/init_ember.py /mnt/e/malware/ember-master/qy-extracted_features/dike -m -t

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值