search

packing-box / docker-packing-box

Docker image gathering packers and tools for making datasets of packed executables and training machine learning models for packing detection
GNU General Public License v3.0
49 stars 10 forks source link

Visualization error with infogain-compare #137

Closed jramhani closed 3 months ago

jramhani commented 3 months ago

dataset plot infogain-compare dataset_example1 --datasets dataset_example2 --max-features 10

00:00:01.745 [INFO] Computing features...
100% ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 131/131 samples • 0:00:03 • 0:00:00 • dataset_example1
00:00:05.366 [INFO] Computing features...
100% ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 131/131 samples • 0:00:03 • 0:00:00 • dataset_example2
00:00:08.801 [INFO] Preparing plot data...
00:00:09.150 [INFO] Computing features...
100% ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 131/131 samples • 0:00:02 • 0:00:00 • dataset_example1
Traceback (most recent call last):
  File "/home/user/.opt/tools/dataset", line 215, in <module>
    getattr(ds, args.command)(**vars(args))
  File "/home/user/.local/lib/python3.11/site-packages/pbox/core/dataset/__init__.py", line 680, in plot
    _PLOTS[subcommand](self, **kw)
  File "/home/user/.local/lib/python3.11/site-packages/pbox/helpers/figure.py", line 83, in _wrapper
    imgs = f(*a, **kw)
           ^^^^^^^^^^^
  File "/home/user/.local/lib/python3.11/site-packages/pbox/core/dataset/visualization.py", line 286, in _information_gain_comparison_heatmap
    feature = filter_features(dataset, feature or "*")
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/user/.local/lib/python3.11/site-packages/pbox/helpers/data.py", line 74, in filter_features
    for f in dataset._features.keys():
             ^^^^^^^^^^^^^^^^^
AttributeError: 'Dataset' object has no attribute '_features'