evilsocket / ergo

🧠 A tool that makes AI easier.
Other
295 stars 36 forks source link

Error while encode #6

Closed 3vangel1st closed 5 years ago

3vangel1st commented 5 years ago

Getting the below error when I try to run the encode command to classify my malicious and clean files:

:~$ ergo encode malware/ data/ --output dataset.csv [2019-05-29 03:41:35,405] (INFO) loading project /home/remnux/malware ... [2019-05-29 03:41:35,410] (INFO) building model for training ... [2019-05-29 03:41:35,449] (WARNING) From /usr/local/lib/python3.4/dist-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version. Instructions for updating: Colocations handled automatically by placer. [2019-05-29 03:41:35,465] (WARNING) From /usr/local/lib/python3.4/dist-packages/keras/backend/tensorflow_backend.py:3445: calling dropout (from tensorflow.python.ops.nn_ops) with keep_prob is deprecated and will be removed in a future version. Instructions for updating: Please use rate instead of keep_prob. Rate should be set to rate = 1 - keep_prob. [2019-05-29 03:41:35,509] (INFO) using auto labeling [2019-05-29 03:41:35,510] (INFO) enumerating data/clean ... [2019-05-29 03:41:35,510] (INFO) enumerating data/malicious ... [2019-05-29 03:41:35,510] (INFO) labeling 0 files ... [2019-05-29 03:41:35,510] (INFO) starting 8 workers for encoding [2019-05-29 03:41:35,522] (INFO) encoding 0 inputs to dataset.csv ...

Process Process-1: Traceback (most recent call last): File "/usr/lib/python3.4/multiprocessing/process.py", line 254, in _bootstrap self.run() File "/usr/lib/python3.4/multiprocessing/process.py", line 93, in run self._target(*self._args, **self._kwargs) File "/usr/local/lib/python3.4/dist-packages/ergo/actions/encode.py", line 86, in appender on_progress(done, total) File "/usr/local/lib/python3.4/dist-packages/ergo/actions/encode.py", line 70, in on_progress perc = done / total ZeroDivisionError: division by zero

evilsocket commented 5 years ago

[2019-05-29 03:41:35,522] (INFO) encoding 0 inputs to dataset.csv ...

you are using it wrong

evilsocket commented 5 years ago

the syntax is:

ergo encode MODEL_PATH DATA_PATH --output dataset.csv --filter "*.exe"

Where:

3vangel1st commented 5 years ago

Many thanks for your feedback - will try it again