Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,211 Bytes
35e2073 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
#!/usr/bin/env python2
import os
# import yaml
import numpy as np
def compute_statistics(data_vec):
stats = dict()
if len(data_vec) > 0:
stats['rmse'] = float(
np.sqrt(np.dot(data_vec, data_vec) / len(data_vec)))
stats['mean'] = float(np.mean(data_vec))
stats['median'] = float(np.median(data_vec))
stats['std'] = float(np.std(data_vec))
stats['min'] = float(np.min(data_vec))
stats['max'] = float(np.max(data_vec))
stats['num_samples'] = int(len(data_vec))
else:
stats['rmse'] = 0
stats['mean'] = 0
stats['median'] = 0
stats['std'] = 0
stats['min'] = 0
stats['max'] = 0
stats['num_samples'] = 0
return stats
# def update_and_save_stats(new_stats, label, yaml_filename):
# stats = dict()
# if os.path.exists(yaml_filename):
# stats = yaml.load(open(yaml_filename, 'r'), Loader=yaml.FullLoader)
# stats[label] = new_stats
#
# with open(yaml_filename, 'w') as outfile:
# outfile.write(yaml.dump(stats, default_flow_style=False))
#
# return
#
#
# def compute_and_save_statistics(data_vec, label, yaml_filename):
# new_stats = compute_statistics(data_vec)
# update_and_save_stats(new_stats, label, yaml_filename)
#
# return new_stats
#
#
# def write_tex_table(list_values, rows, cols, outfn):
# '''
# write list_values[row_idx][col_idx] to a table that is ready to be pasted
# into latex source
#
# list_values is a list of row values
#
# The value should be string of desired format
# '''
#
# assert len(rows) >= 1
# assert len(cols) >= 1
#
# with open(outfn, 'w') as f:
# # write header
# f.write(' & ')
# for col_i in cols[:-1]:
# f.write(col_i + ' & ')
# f.write(' ' + cols[-1]+'\n')
#
# # write each row
# for row_idx, row_i in enumerate(list_values):
# f.write(rows[row_idx] + ' & ')
# row_values = list_values[row_idx]
# for col_idx in range(len(row_values) - 1):
# f.write(row_values[col_idx] + ' & ')
# f.write(' ' + row_values[-1]+' \n')
|