amide-models / jsonshift.py
Samarth Naik
added jsonshift.py
1c9f2d5
import csv
import json
with open('network_logs.csv', 'r') as f:
reader = csv.DictReader(f)
data = []
for row in reader:
row['src_port'] = int(row['src_port'])
row['dst_port'] = int(row['dst_port'])
row['packet_size'] = int(row['packet_size'])
row['tcp_flags'] = int(row['tcp_flags'])
row['seq'] = int(row['seq'])
row['ack'] = int(row['ack'])
row['window'] = int(row['window'])
data.append(row)
payload = {'model_type': 'lightGBM', 'file': data}
with open('payload.json', 'w') as f:
json.dump(payload, f)