def write_weka_file_for_each_user(data, attributes): all_users = {} for row in data: userid = row[len(row) - 1] # User ID is the last element in list # Get User data from row and add to user dictionary # try to get existing key, and append to the existing list in value try: user_list = all_users[userid] user_list.append(row) all_users[userid] = user_list except KeyError: # If doesn't exist, create the list user_list = [row] all_users[userid] = user_list for user, userdata in all_users.items(): weka_data = { 'description': 'Data for ' + user, 'relation': 'SensorRecordings', 'attributes': attributes, 'data': userdata, } # Write Weka formatted file for entire cohort f = open(out_dir[0] + '/' + user + '.arff', 'w') f.write(arff.dumps(weka_data)) f.close()
def write_weka_file_for_cohort(data, attributes): weka_data = { 'description': '', 'relation': 'sensors', 'attributes': attributes, 'data': data, } f = open(out_dir[0] + '/cohort.arff', 'w') f.write(arff.dumps(weka_data)) f.close()