import torch


def get_obs_y_dict(select_pairs, x_a, x_s):
    """
    
    Input
    -------
    select_pairs: pairs of (alpha, beta) selected
    x_a: array holding avoidance count for all dogs & all trials, example for 30 dogs & 25 trials, shaped (30, 25)
    x_s: array holding shock count for all dogs & all trials, example for 30 dogs & 25 trials, shaped (30, 25)
    
    Output
    -------
    
    Outputs a dictionary with tuple of alpha, beta as key & observerd values of y corresponding to alpha, beta in key
    
    """
    y_dict = {}
    for alpha, beta in select_pairs:# pair of alpha, beta
        y_dict[(alpha, beta)] = torch.exp(alpha*x_a + beta* x_s)
    
    return y_dict