Open asafnadler opened 6 years ago
@asafnadler I've just checked and it seems to work okay for me. I have labels correctly assigned. Do you have not correct labels for malicious domain?
In data.py:
domains += get_alexa(len(domains))
labels += ['benign']*len(domains)
change to:
pre_len = len(domains)
domains += get_alexa(len(domains))
labels += ['benign']*pre_len
Use previous domain list length to generate labels with same length, or just divide latter length by 2.
The function gen_data takes the count of malicious+benign domains and adds this number of labels to the malicious labels.