学习笔记（十七）：朴素贝叶斯和多层感知机检测Linux后门

最新推荐文章于 2024-06-18 08:24:53 发布

林咚咚

最新推荐文章于 2024-06-18 08:24:53 发布

阅读量590

点赞数

CC 4.0 BY-SA版权

分类专栏： python web安全

本文链接：https://blog.youkuaiyun.com/weixin_39878297/article/details/84307686

1.数据集使用ADFA-LD数据集

def load_all_files():
    import glob
    x=[]
    y=[]
    files=glob.glob("ADFA-LD/Attack_Data_Master/*/*")
    for file in files:
        with open(file) as f:
            lines=f.readlines()
        x.append(" ".join(lines))
        y.append(1)
    print("Load black data %d" % len(x))
    files=glob.glob("ADFA-LD/Training_Data_Master/*")
    for file in files:
        with open(file) as f:
            lines=f.readlines()
        x.append(" ".join(lines))
        y.append(0)
    print("Load full data %d" % len(x))

    return x,y

2.使用N-Gram和TF-IDF提取特征

def get_feature_wordbag():
    max_features=1000
    x,y=load_all_files()
    vectorizer = CountVectorizer(
                                 ngram_range=(3, 3),
                                 token_pattern=r'\b\d+\b',
                                 decode_error='ignore',
                                 strip_accents='ascii',