python图片自动校正流量_python网络流量异常检测

import tensorflow as tf

from sklearn.model_selection import train_test_split

from tensorflow.keras.models import Sequential

import numpy,time

from collections import Counter

from tensorflow.keras import layers

from sklearn.metrics import confusion_matrix

from sklearn.svm import SVC

from sklearn.naive_bayes import GaussianNB

from sklearn.neighbors import KNeighborsClassifier

from sklearn.tree import DecisionTreeClassifier

from sklearn import metrics

import matplotlib.pyplot as plt

def handleProtocol(input):

protocol_list=['tcp','udp','icmp']

if input in protocol_list:

return protocol_list.index(input)

def handleService(input):

service_list=['aol','auth','bgp','courier','csnet_ns','ctf','daytime','discard','domain','domain_u',

'echo','eco_i','ecr_i','efs','exec','finger','ftp','ftp_data','gopher','harvest','hostnames',

'http','http_2784','http_443','http_8001','imap4','IRC','iso_tsap','klogin','kshell','ldap',

'link','login','mtp','name','netbios_dgm','netbios_ns','netbios_ssn','netstat','nnsp','nntp',

'ntp_u','other','pm_dump','pop_2','pop_3','printer','private','red_i','remote_job','rje','shell',

'smtp','sql_net','ssh','sunrpc','supdup','systat','telnet','tftp_u','tim_i','time','urh_i','urp_i',

'uucp','uucp_path','vmnet','whois','X11','Z39_50']

if input in service_list:

return service_list.index(input)

def handleFlag(input):

flag_list=['OTH','REJ','RSTO','RSTOS0','RSTR','S0','S1','S2','S3','SF','SH']

if input in flag_list:

return flag_list.index(input)

def handleLabel(input):

#多了一个’udpstorm标签(5分类，2分类)

try:

label_dict ={'normal':0}

dos = dict(zip(['back','neptune','