Browse Source

add base utils for predictor

main
MahsaYazdani 2 years ago
parent
commit
855c4b8faf
1 changed files with 10 additions and 0 deletions
  1. 10
    0
      predictor/model/utils.py

+ 10
- 0
predictor/model/utils.py View File

@@ -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')

Loading…
Cancel
Save