From 3fa99ce033177ef2a2babfca504bfdb810128db6 Mon Sep 17 00:00:00 2001
From: Zhiying Xu <33195129+xuzhiying9510@users.noreply.github.com>
Date: Tue, 6 Jul 2021 03:56:01 -0400
Subject: [PATCH] Update background.py

---
 botgen/background.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/botgen/background.py b/botgen/background.py
index f38bd99..cc71954 100644
--- a/botgen/background.py
+++ b/botgen/background.py
@@ -68,8 +68,8 @@ def prepare_background(f, dst_dir, dst_name, graph_id, start_time, stop_time):
     df["dstIP"] = df["dstIP"].apply(ip2int)
     
     #aggregate nodes, build dictionary
-    df['srcIP'] = df['srcIP'].apply(lambda x: x >> 8)#
-    df['dstIP'] = df['dstIP'].apply(lambda x: x >> 8)#
+    df['srcIP'] = df['srcIP'].apply(lambda x: x >> 12)# use ip/20 in our experiment, feel free to change
+    df['dstIP'] = df['dstIP'].apply(lambda x: x >> 12)# use ip/20 in our experiment, feel free to change
     df = df.drop_duplicates()
     
     #renumber into 0, 1, 2, ..