File size: 588 Bytes
1c9f2d5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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)