please download dataset to data folder before running the code.

also put .csv under data folder

note that training dataset should be under data/train/ instead of data/train/train/, same for testing data

python train.py --model 'your model' --lr 'your learning rate' --batch_size 'your batch size'

for new model, run python train_embedding --model 'MalwareNet' --lr 'your learning rate' --batch_size 'your batch size'

https://docs.google.com/document/d/1rwosWVQol3FvX_DiEOCTYrzg1lzIzxNbc93R3-xYqiQ/edit?usp=sharing