@@ -6,6 +6,16 @@ import torch | |||
import json | |||
import random | |||
def calc_stat(numbers): | |||
mu = sum(numbers) / len(numbers) | |||
sigma = (sum([(x - mu) ** 2 for x in numbers]) / len(numbers)) ** 0.5 | |||
return mu, sigma | |||
def conf_inv(mu, sigma, n): | |||
delta = 2.776 * sigma / (n ** 0.5) # 95% | |||
return mu - delta, mu + delta | |||
def arg_min(lst): | |||
m = float('inf') |