Training.py error
Hi,
When I tried to execute train.py I encounter error as follows
D:\x_malwdetc\x_mawdetc\src\deep_malware_detection>python train.py --benign_dir="D:\x_malwdetc\x_mawdetc\src\bin\raw\dll-headersiz" --malware_dir="D:\x_malwdetc\x_mawdetc\src\bin\raw\dasmalwerk-headersiz"
Traceback (most recent call last):
File "D:\x_malwdetc\x_mawdetc\src\deep_malware_detection\train.py", line 4, in
Would you help me to resolve this issue ? Thanks in advance
The error is not coming from this library or code. Can you double check that you are able to import PyTorch in your virtual environment?
>>> import torch
By the way, in files uploading to the neural network, which parameter is considered in learning process to determine whether it is benignware or malware ? For instance is it done by PE format or something else ?
It's the PE format. The README contains some references I used for the implementation.