Skip to content

Commit

Permalink
add screen shoot
Browse files Browse the repository at this point in the history
  • Loading branch information
PENGZhaoqing committed Nov 29, 2016
1 parent 1a3eac8 commit 2969959
Show file tree
Hide file tree
Showing 11 changed files with 11 additions and 13 deletions.
2 changes: 1 addition & 1 deletion CART/CART_Trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ class Trainer:

def train(self, training_set, training_target, fea_index):

clf = tree.DecisionTreeClassifier(criterion="gini", max_depth=7, min_samples_leaf=5)
clf = tree.DecisionTreeClassifier(criterion="gini", max_depth=7, min_samples_leaf=50)
clf = clf.fit(training_set, training_target)

class_names = np.unique([str(i) for i in training_target])
Expand Down
Binary file modified CART/CART_Trainer.pyc
Binary file not shown.
Binary file modified CART/output/CART.pkl
Binary file not shown.
Binary file modified CART/output/tree-vis.pdf
Binary file not shown.
4 changes: 2 additions & 2 deletions MLP/MLP_Trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,11 +66,11 @@ def one_hot_encoding(self, dataset, datatarget, T_len):
pca = PCA(n_components=20)
pca.fit(data_set)

print data_set.shape

data_set = pca.transform(data_set)
test_set = pca.transform(test_set)

print data_set.shape

scaler = StandardScaler()
scaler.fit(data_set)

Expand Down
Binary file modified MLP/output/MLP.pkl
Binary file not shown.
16 changes: 8 additions & 8 deletions Mongo_Con.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,11 +44,11 @@ def import_test_data(self, test_file):
self.db.test_data.insert_one({"test_set": dic})

def CART_fetch_data(self):
training_cursor = self.db.training_data.find({"training_set.src_bytes": {"$gt": 10000}})
test_cursor = self.db.test_data.find({"test_set.src_bytes": {"$gt": 10000}})
# training_cursor = self.db.training_data.find({"training_set.src_bytes": {"$gt": 10000}})
# test_cursor = self.db.test_data.find({"test_set.src_bytes": {"$gt": 10000}})

# training_cursor = self.db.training_data.find()
# test_cursor = self.db.test_data.find()
training_cursor = self.db.training_data.find()
test_cursor = self.db.test_data.find()

cursor = training_cursor.sort('training_set.type', pymongo.ASCENDING)

Expand Down Expand Up @@ -81,11 +81,11 @@ def CART_fetch_data(self):
return np.array(dataset), np.array(dataTarget), training_len

def MLP_fetch_data(self):
training_cursor = self.db.training_data.find({"training_set.src_bytes": {"$gt": 10000}})
test_cursor = self.db.test_data.find({"test_set.src_bytes": {"$gt": 100000}})
# training_cursor = self.db.training_data.find({"training_set.src_bytes": {"$gt": 10000}})
# test_cursor = self.db.test_data.find({"test_set.src_bytes": {"$gt": 100000}})

# training_cursor = self.db.training_data.find()
# test_cursor = self.db.test_data.find()
training_cursor = self.db.training_data.find()
test_cursor = self.db.test_data.find()

cursor = training_cursor.sort('training_set.type', pymongo.ASCENDING)

Expand Down
Binary file modified Mongo_Con.pyc
Binary file not shown.
Binary file added Snip20161130_3.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1 change: 0 additions & 1 deletion data/kddcup.data_10_percent.txt
Original file line number Diff line number Diff line change
Expand Up @@ -485795,7 +485795,6 @@
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,217,20,1.00,1.00,0.00,0.00,0.09,0.06,0.00,255,20,0.08,0.06,0.00,0.00,1.00,1.00,0.00,0.00,2
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,227,13,1.00,1.00,0.00,0.00,0.06,0.05,0.00,255,13,0.05,0.06,0.00,0.00,1.00,1.00,0.00,0.00,2
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,237,7,1.00,1.00,0.00,0.00,0.03,0.05,0.00,255,7,0.03,0.06,0.00,0.00,1.00,1.00,0.00,0.00,2
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,00,tcp,http,SF,334,1684,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,1,9,0.00,0.00,0.00,0
0,tcp,http,SF,332,1076,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,11,12,0.00,0.00,0.00,0.00,1.00,0.00,0.17,10,10,1.00,0.00,0.10,0.00,0.00,0.00,0.00,0.00,0
0,tcp,http,SF,344,2280,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,8,8,0.00,0.00,0.00,0.00,1.00,0.00,0.00,8,20,1.00,0.00,0.12,0.10,0.00,0.00,0.00,0.00,0
0,tcp,http,SF,180,1929,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,3,5,0.00,0.00,0.00,0.00,1.00,0.00,0.40,18,30,1.00,0.00,0.06,0.07,0.00,0.00,0.00,0.00,0
Expand Down
1 change: 0 additions & 1 deletion raw/kddcup.data_10_percent.txt
Original file line number Diff line number Diff line change
Expand Up @@ -485795,7 +485795,6 @@
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,217,20,1.00,1.00,0.00,0.00,0.09,0.06,0.00,255,20,0.08,0.06,0.00,0.00,1.00,1.00,0.00,0.00,neptune.
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,227,13,1.00,1.00,0.00,0.00,0.06,0.05,0.00,255,13,0.05,0.06,0.00,0.00,1.00,1.00,0.00,0.00,neptune.
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,237,7,1.00,1.00,0.00,0.00,0.03,0.05,0.00,255,7,0.03,0.06,0.00,0.00,1.00,1.00,0.00,0.00,neptune.
0,tcp,private,S0,0,0,0,0,0,0,0,0,0,0,00,tcp,http,SF,334,1684,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,1,9,0.00,0.00,0.00,0.00,1.00,0.00,0.33,0,0,0.00,0.00,0.00,0.00,0.00,0.00,0.00,0.00,normal.
0,tcp,http,SF,332,1076,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,11,12,0.00,0.00,0.00,0.00,1.00,0.00,0.17,10,10,1.00,0.00,0.10,0.00,0.00,0.00,0.00,0.00,normal.
0,tcp,http,SF,344,2280,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,8,8,0.00,0.00,0.00,0.00,1.00,0.00,0.00,8,20,1.00,0.00,0.12,0.10,0.00,0.00,0.00,0.00,normal.
0,tcp,http,SF,180,1929,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,3,5,0.00,0.00,0.00,0.00,1.00,0.00,0.40,18,30,1.00,0.00,0.06,0.07,0.00,0.00,0.00,0.00,normal.
Expand Down

0 comments on commit 2969959

Please sign in to comment.