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

`TypeError` on `model train --n-jobs` #94

Closed AlexVanMechelen closed 8 months ago

AlexVanMechelen commented 8 months ago

Steps to reproduce

dataset make test -n 300 -f PE
model train test -A rf --n-jobs 32

Traceback

┌──[user@packing-box]──[/mnt/share]──[main|✓]────────                          ────[172.17.0.2]──[16:50:04]────
$ model train test -A svm --n-jobs 32
00:00:01.080 [INFO] Selected algorithm: Support Vector Machine
Traceback (most recent call last):
  File "/home/user/.opt/tools/model", line 114, in <module>
    getattr(Model(**vars(args)), args.command)(**vars(args))
  File "/home/user/.local/lib/python3.11/site-packages/pbox/core/model/__init__.py", line 471, in train
    if n_jobs > n_cpu:
       ^^^^^^^^^^^^^^
TypeError: '>' not supported between instances of 'str' and 'int'