Change extract features function to return capture source
@@ -184,4 +184,4 @@ def extract_features(session_dict, capture_source=None, max_port=1024):
extra_features[3] = num_icmp_sess/num_sessions
feature_vector = np.concatenate((num_port_sess, extra_features), axis=0)
- return feature_vector
+ return feature_vector, capture_source
