SimulatingVariants/scoring_functions.py at master · roybens/SimulatingVariants · GitHub

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
import eval_helper as eh
'''
This file contains various scoring functions that can be used in the evaluator's
calc_rmse function.
'''

def calc_rmse_standard(target_data):
    '''
    Calculates the inactivation, activation, recovery, and tau0 rmse
    in that order compared to the current state of the neuron.

    params:
        target_data: a dictionary of the experimental data in standard format
    returns:
        List of rmse values in order
    '''
    try:
        sim_data = eh.gen_sim_data()
    except ZeroDivisionError: #catch error to prevent bad individuals from halting run
        print("ZeroDivisionError when generating sim_data, returned infinity.")
        sim_data =None
        return (1000,1000,1000,1000)
    try:
        inds = target_data["inact sig inds"]
        squared_diffs = [(target_data['inact'][i]-sim_data['inact'][i])**2 for i in inds]
        inact_rmse = (sum(squared_diffs)/len(inds))**.5

        inds = target_data["act sig inds"]
        squared_diffs = [(target_data['act'][i]-sim_data['act'][i])**2 for i in inds]
        act_rmse = (sum(squared_diffs)/len(inds))**.5

        inds = target_data["recov sig inds"]
        squared_diffs = [(target_data['recov'][i]-sim_data['recov'][i])**2 for i in inds]
        recov_rmse = (sum(squared_diffs)/len(inds))**.5

        tau_rmse = ((target_data['tau0']-sim_data['tau0'])**2)**.5
    except OverflowError:
        print('OverflowError when calculating rmse, returned infinity.')
        return (1000, 1000, 1000, 1000)
    return [inact_rmse, act_rmse, recov_rmse, tau_rmse]

def calc_rmse_sans_tau(target_data):
    '''
    Calculates the inactivation, activation, and recovery rmse
    in that order

    params:
        target_data: a dictionary of the experimental data in standard format
    returns:
        List of rmse values in order
    '''
    try:
        sim_data = eh.gen_sim_data()
    except ZeroDivisionError: #catch error to prevent bad individuals from halting run
        print("ZeroDivisionError when generating sim_data, returned infinity.")
        sim_data =None
        return (1000,1000,1000)
    try:
        inds = target_data["inact sig inds"]
        squared_diffs = [(target_data['inact'][i]-sim_data['inact'][i])**2 for i in inds]
        inact_rmse = (sum(squared_diffs)/len(inds))**.5

        inds = target_data["act sig inds"]
        squared_diffs = [(target_data['act'][i]-sim_data['act'][i])**2 for i in inds]
        act_rmse = (sum(squared_diffs)/len(inds))**.5

        inds = target_data["recov sig inds"]
        squared_diffs = [(target_data['recov'][i]-sim_data['recov'][i])**2 for i in inds]
        recov_rmse = (sum(squared_diffs)/len(inds))**.5

    except OverflowError:
        print('OverflowError when calculating rmse, returned infinity.')
        return (1000, 1000, 1000)
    return [inact_rmse, act_rmse, recov_rmse]